Python给你的头像加上圣诞帽


Posted in Python onJanuary 04, 2018

引言

随着圣诞的到来,大家纷纷@官方微信给自己的头像加上一顶圣诞帽。当然这种事情用很多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.shape[1],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给你的头像加上圣诞帽

下载:完整代码。

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

Python 相关文章推荐
python装饰器实例大详解
Oct 25 Python
python调用摄像头显示图像的实例
Aug 03 Python
详解pandas使用drop_duplicates去除DataFrame重复项参数
Aug 01 Python
Python多线程爬取豆瓣影评API接口
Oct 22 Python
Python实现socket非阻塞通讯功能示例
Nov 06 Python
python的slice notation的特殊用法详解
Dec 27 Python
python 实现字符串下标的输出功能
Feb 13 Python
使用Python发现隐藏的wifi
Mar 04 Python
python os模块在系统管理中的应用
Jun 22 Python
django美化后台django-suit的安装配置操作
Jul 12 Python
Python自动化办公Excel模块openpyxl原理及用法解析
Nov 05 Python
Python OpenCV实现传统图片格式与base64转换
Jun 13 Python
Python编程实现线性回归和批量梯度下降法代码实例
Jan 04 #Python
Python语言描述随机梯度下降法
Jan 04 #Python
微信小程序跳一跳游戏 python脚本跳一跳刷高分技巧
Jan 04 #Python
python使用xpath中遇到:到底是什么?
Jan 04 #Python
python使用itchat库实现微信机器人(好友聊天、群聊天)
Jan 04 #Python
python实现微信跳一跳辅助工具步骤详解
Jan 04 #Python
Python中实现最小二乘法思路及实现代码
Jan 04 #Python
You might like
PHP 中使用explode()函数切割字符串为数组的示例
2017/05/06 PHP
php多进程中的阻塞与非阻塞操作实例分析
2020/03/04 PHP
ThinkPHP3.1.2 使用cli命令行模式运行的方法
2020/04/14 PHP
jQueryPad 实用的jQuery测试工具(支持IE,chrome,FF)
2010/05/22 Javascript
javascript操作referer详细解析
2014/03/10 Javascript
javascript实现类似百度分享功能的方法
2015/07/27 Javascript
Jquery ajax 同步阻塞引起的UI线程阻塞问题
2015/11/17 Javascript
jquery仿QQ登录账号选择下拉框效果
2016/03/22 Javascript
用原生JS对AJAX做简单封装的实例代码
2016/07/13 Javascript
记一次webpack3升级webpack4的踩坑经历
2018/06/12 Javascript
js代码规范之Eslint安装与配置详解
2018/09/08 Javascript
解决ng-repeat产生的ng-model中取不到值的问题
2018/10/02 Javascript
js canvas画布实现高斯模糊效果
2018/11/27 Javascript
仿ElementUI实现一个Form表单的实现代码
2019/04/23 Javascript
微信小程序返回上一页传参并刷新过程解析
2019/12/13 Javascript
[02:14]DOTA2英雄基础教程 修补匠
2013/12/23 DOTA
Python Tkinter GUI编程入门介绍
2015/03/10 Python
Python教程之全局变量用法
2016/06/27 Python
Python与Java间Socket通信实例代码
2017/03/06 Python
深入理解python中的atexit模块
2017/03/07 Python
python3+mysql查询数据并通过邮件群发excel附件
2018/02/24 Python
python远程连接服务器MySQL数据库
2018/07/02 Python
使用memory_profiler监测python代码运行时内存消耗方法
2018/12/03 Python
Python中请不要再用re.compile了
2019/06/30 Python
pytorch中 gpu与gpu、gpu与cpu 在load时相互转化操作
2020/05/25 Python
学生如何注册Pycharm专业版以及pycharm的安装
2020/09/24 Python
美国轮胎网站:Priority Tire
2018/11/28 全球购物
Genny意大利官网:意大利高级时装品牌
2020/04/15 全球购物
DataReader和DataSet的异同
2014/12/31 面试题
安全生产活动月方案
2014/03/09 职场文书
cf收人广告词大全
2014/03/14 职场文书
2014国庆65周年领导讲话稿(3篇)
2014/09/21 职场文书
交通事故赔偿协议书怎么写
2014/10/04 职场文书
爱护环境卫生倡议书
2015/04/29 职场文书
2016习总书记系列重要讲话心得体会
2016/01/15 职场文书
优秀范文:《但愿人长久》教学反思3篇
2019/10/24 职场文书