使用Python给头像戴上圣诞帽的图像操作过程解析


Posted in Python onSeptember 20, 2019

前言

随着圣诞的到来,大家纷纷@官方微信给自己的头像加上一顶圣诞帽。当然这种事情用很多P图软件都可以做到。但是作为一个学习图像处理的技术人,还是觉得我们有必要写一个程序来做这件事情。而且这完全可以作为一个练手的小项目,工作量不大,而且很有意思。

用到的工具

  • OpenCV(毕竟我们主要的内容就是OpenCV...)
  • dlib(dlib的人脸检测比OpenCV更好用,而且dlib有OpenCV没有的关键点检测。)

用到的语言为Python。但是完全可以改成C++版本,时间有限,就不写了。有兴趣的小伙伴可以拿来练手。

流程

一、素材准备

首先我们需要准备一个圣诞帽的素材,格式最好为PNG,因为PNG的话我们可以直接用Alpha通道作为掩膜使用。我们用到的圣诞帽如下图:

使用Python给头像戴上圣诞帽的图像操作过程解析

我们通过通道分离可以得到圣诞帽图像的alpha通道。代码如下:

r,g,b,a = cv2.split(hat_img) 
 rgb_hat = cv2.merge((r,g,b))
 cv2.imwrite("hat_alpha.jpg",a)

为了能够与rgb通道的头像图片进行运算,我们把rgb三通道合成一张rgb的彩色帽子图。Alpha通道的图像如下图所示。

使用Python给头像戴上圣诞帽的图像操作过程解析

二、人脸检测与人脸关键点检测

我们用下面这张图作为我们的测试图片。

使用Python给头像戴上圣诞帽的图像操作过程解析

下面我们用dlib的正脸检测器进行人脸检测,用dlib提供的模型提取人脸的五个关键点。代码如下:

# dlib人脸关键点检测器 predictor_path = "shape_predictor_5_face_landmarks.dat" predictor = dlib.shape_predictor(predictor_path)

 # dlib正脸检测器
 detector = dlib.get_frontal_face_detector()

 # 正脸检测
 dets = detector(img, 1)

 # 如果检测到人脸
 if len(dets)>0: 
   for d in dets:
     x,y,w,h = d.left(),d.top(), d.right()-d.left(), d.bottom()-d.top()
     # x,y,w,h = faceRect 
     cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2,8,0)

     # 关键点检测,5个关键点
     shape = predictor(img, d)
     for point in shape.parts():
       cv2.circle(img,(point.x,point.y),3,color=(0,255,0))

     cv2.imshow("image",img)
     cv2.waitKey()

这部分效果如下图:

使用Python给头像戴上圣诞帽的图像操作过程解析

三、调整帽子大小

我们选取两个眼角的点,求中心作为放置帽子的x方向的参考坐标,y方向的坐标用人脸框上线的y坐标表示。然后我们根据人脸检测得到的人脸的大小调整帽子的大小,使得帽子大小合适。

# 选取左右眼眼角的点 point1 = shape.part(0) point2 = shape.part(2)
     # 求两点中心
     eyes_center = ((point1.x+point2.x)//2,(point1.y+point2.y)//2)
     # cv2.circle(img,eyes_center,3,color=(0,255,0)) 
     # cv2.imshow("image",img)
     # cv2.waitKey()
     # 根据人脸大小调整帽子大小
     factor = 1.5
     resized_hat_h = int(round(rgb_hat.shape[0]*w/rgb_hat.shape[1]*factor))
     resized_hat_w = int(round(rgb_hat.shape[1]*w/rgb_hat.shape[1]*factor))
     if resized_hat_h > y:
       resized_hat_h = y-1
     # 根据人脸大小调整帽子大小
     resized_hat = cv2.resize(rgb_hat,(resized_hat_w,resized_hat_h))

四、提取帽子和需要添加帽子的区域

按照之前所述,去Alpha通道作为mask。并求反。这两个mask一个用于把帽子图中的帽子区域取出来,一个用于把人物图中需要填帽子的区域空出来。后面你将会看到。

用alpha通道作为mask

mask = cv2.resize(a,(resized_hat_w,resized_hat_h))
     mask_inv = cv2.bitwise_not(mask)

从原图中取出需要添加帽子的区域,这里我们用的是位运算操作。

# 帽子相对与人脸框上线的偏移量 dh = 0 dw = 0 # 原图ROI # bg_roi = img[y+dh-resized_hat_h:y+dh, x+dw:x+dw+resized_hat_w] bg_roi = img[y+dh-resized_hat_h:y+dh,(eyes_center[0]-resized_hat_w//3):(eyes_center[0]+resized_hat_w//3*2)]
     # 原图ROI中提取放帽子的区域
     bg_roi = bg_roi.astype(float)
     mask_inv = cv2.merge((mask_inv,mask_inv,mask_inv))
     alpha = mask_inv.astype(float)/255

     # 相乘之前保证两者大小一致(可能会由于四舍五入原因不一致)
     alpha = cv2.resize(alpha,(bg_roi.shape[1],bg_roi.shape[0]))
     # print("alpha size: ",alpha.shape)
     # print("bg_roi size: ",bg_roi.shape)
     bg = cv2.multiply(alpha, bg_roi)
     bg = bg.astype('uint8')

这是的背景区域(bg)如下图所示。可以看到,刚好是需要填充帽子的区域缺失了。

使用Python给头像戴上圣诞帽的图像操作过程解析

然后我们提取帽子区域。

# 提取帽子区域 hat = cv2.bitwise_and(resized_hat,resized_hat,mask = mask)

提取得到的帽子区域如下图。帽子区域正好与上一个背景区域互补。

使用Python给头像戴上圣诞帽的图像操作过程解析

五、添加圣诞帽

最后我们把两个区域相加。再放回到原图中去,就可以得到我们想要的圣诞帽图了。这里需要注意的就是,相加之前resize一下保证两者大小一致,因为可能会由于四舍五入原因不一致。

# 相加之前保证两者大小一致(可能会由于四舍五入原因不一致) hat = cv2.resize(hat,(bg_roi.shape1,bg_roi.shape[0])) # 两个ROI区域相加 add_hat = cv2.add(bg,hat) # cv2.imshow("add_hat",add_hat)
     # 把添加好帽子的区域放回原图
     img[y+dh-resized_hat_h:y+dh,(eyes_center[0]-resized_hat_w//3):(eyes_center[0]+resized_hat_w//3*2)] = add_hat

最后我们得到的效果图如下所示。

使用Python给头像戴上圣诞帽的图像操作过程解析

源码地址:https://github.com/LiuXiaolong19920720/Add-Christmas-Hat

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
用python实现对比两张图片的不同
Feb 05 Python
Python简单计算给定某一年的某一天是星期几示例
Jun 27 Python
Python求两个圆的交点坐标或三个圆的交点坐标方法
Nov 07 Python
pygame游戏之旅 python和pygame安装教程
Nov 20 Python
Python中@property的理解和使用示例
Jun 11 Python
Python使用QQ邮箱发送邮件报错smtplib.SMTPAuthenticationError
Dec 20 Python
利用django model save方法对未更改的字段依然进行了保存
Mar 28 Python
Python使用matplotlib绘制圆形代码实例
May 27 Python
python map比for循环快在哪
Sep 21 Python
Python实现区域填充的示例代码
Feb 03 Python
Python上下文管理器Content Manager
Jun 26 Python
python 离散点图画法的实现
Apr 01 Python
Python 函数用法简单示例【定义、参数、返回值、函数嵌套】
Sep 20 #Python
Python 50行爬虫抓取并处理图灵书目过程详解
Sep 20 #Python
使用python获取邮箱邮件的设置方法
Sep 20 #Python
Python中的上下文管理器相关知识详解
Sep 19 #Python
Python Opencv提取图片中某种颜色组成的图形的方法
Sep 19 #Python
pandas read_excel()和to_excel()函数解析
Sep 19 #Python
python openvc 裁剪、剪切图片 提取图片的行和列
Sep 19 #Python
You might like
PHP 和 MySQL 基础教程(三)
2006/10/09 PHP
PHP错误机制知识汇总
2016/03/24 PHP
Yii的Srbac插件用法详解
2016/07/14 PHP
PHP chunk_split()函数讲解
2019/02/12 PHP
用YUI做了个标签浏览效果
2007/02/20 Javascript
js跨域和ajax 跨域问题的实现思路
2009/09/05 Javascript
jQuery 使用手册(七)
2009/09/23 Javascript
js Date概念详细介绍
2013/11/22 Javascript
jQuery中的val()示例应用
2014/02/26 Javascript
浅析jQuery中调用ajax方法时在不同浏览器中遇到的问题
2014/06/11 Javascript
使用Sticker.js实现贴纸效果
2015/01/28 Javascript
javascript中SetInterval与setTimeout的定时器用法
2015/08/24 Javascript
bootstrap中模态框、模态框的属性实例详解
2017/02/17 Javascript
JS仿Base.js实现的继承示例
2017/04/07 Javascript
jQuery初级教程之网站品牌列表效果
2017/08/02 jQuery
Vue 2.0学习笔记之Vue中的computed属性
2017/10/16 Javascript
关于jQuery里prev()的简单操作代码
2017/10/27 jQuery
最实用的JS数组函数整理
2017/12/05 Javascript
基于JavaScript实现瀑布流布局
2018/08/15 Javascript
微信小程序自定义单项选择器样式
2019/07/25 Javascript
解决element-ui里的下拉多选框 el-select 时,默认值不可删除问题
2020/08/14 Javascript
python的几种开发工具介绍
2007/03/07 Python
python自动zip压缩目录的方法
2015/06/28 Python
Python之使用adb shell命令启动应用的方法详解
2019/01/07 Python
英国第一蛋白粉品牌:Myprotein
2016/09/14 全球购物
领先的钻石和订婚戒指零售商:Diamonds-USA
2016/12/11 全球购物
威盛公司软件C++工程师笔试题面试题
2012/07/16 面试题
造价工程师个人求职信
2013/09/21 职场文书
行政总经理岗位职责
2013/12/05 职场文书
工作评语大全
2014/04/26 职场文书
群众路线教育实践活动对照检查材料
2014/09/22 职场文书
全国法院系统开展党的群众路线教育实践活动综述(全文)
2014/10/25 职场文书
竞聘报告优秀范文
2014/11/06 职场文书
结婚仪式主持词
2015/06/29 职场文书
技术入股协议书
2016/03/22 职场文书
2016年“我们的节日·中秋节”活动总结
2016/04/05 职场文书