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的源码浅要剖析Python的内存管理
Apr 16 Python
Python实现的十进制小数与二进制小数相互转换功能
Oct 12 Python
PyCharm 常用快捷键和设置方法
Dec 20 Python
PyQt5实现QLineEdit添加clicked信号的方法
Jun 25 Python
pandas删除行删除列增加行增加列的实现
Jul 06 Python
Python中list的交、并、差集获取方法示例
Aug 01 Python
django自带serializers序列化返回指定字段的方法
Aug 21 Python
浅谈Django QuerySet对象(模型.objects)的常用方法
Mar 28 Python
django rest framework使用django-filter用法
Jul 15 Python
Python logging模块原理解析及应用
Aug 13 Python
Python3+PyCharm+Django+Django REST framework配置与简单开发教程
Feb 16 Python
基于PyQt5制作一个群发邮件工具
Apr 08 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
使用Smarty 获取当前日期时间和格式化日期时间的方法详解
2013/06/18 PHP
php实现XML和数组的相互转化功能示例
2017/02/08 PHP
PHP的RSA加密解密方法以及开发接口使用
2018/02/11 PHP
Laravel框架路由设置与使用示例
2018/06/12 PHP
JavaScript省市联动实现代码
2014/02/15 Javascript
flash遮住div问题的正确解决方法
2014/02/27 Javascript
JavaScript数组常用方法
2015/03/02 Javascript
JavaScript仿商城实现图片广告轮播实例代码
2016/02/06 Javascript
JavaScript入门教程之引用类型
2016/05/04 Javascript
浅析JavaScript回调函数应用
2016/05/22 Javascript
JavaScript中的对象和原型(一)
2016/08/12 Javascript
分享JS代码实现鼠标放在输入框上输入框和图片同时更换样式
2016/09/01 Javascript
JS实现重新加载当前页面或者父页面的几种方法
2016/11/30 Javascript
利用Query+bootstrap和js两种方式实现日期选择器
2017/01/10 Javascript
AngularJS constant和value区别详解
2017/02/28 Javascript
Grunt针对静态文件的压缩,版本控制打包的实例讲解
2017/09/29 Javascript
jQuery ajax调用webservice注意事项
2017/10/08 jQuery
详解react-router 4.0 下服务器如何配合BrowserRouter
2017/12/29 Javascript
Vue.set() this.$set()引发的视图更新思考及注意事项
2018/08/30 Javascript
微信小程序左滑删除实现代码实例
2019/09/16 Javascript
[10:42]Team Liquid Vs Newbee
2018/06/07 DOTA
[37:29]完美世界DOTA2联赛PWL S2 LBZS vs Forest 第二场 11.19
2020/11/19 DOTA
Python操作json数据的一个简单例子
2014/04/17 Python
Python 用Redis简单实现分布式爬虫的方法
2017/11/23 Python
Python装饰器模式定义与用法分析
2018/08/06 Python
将string类型的数据类型转换为spark rdd时报错的解决方法
2019/02/18 Python
Python基础之函数基本用法与进阶详解
2020/01/02 Python
解决python 找不到module的问题
2020/02/12 Python
解决Pycharm中恢复被exclude的项目问题(pycharm source root)
2020/02/14 Python
JetBrains PyCharm(Community版本)的下载、安装和初步使用图文教程详解
2020/03/19 Python
欧舒丹比利时官网:L’OCCITANE比利时
2017/04/25 全球购物
洛杉矶健身中心女性专用运动服饰品牌:Marika
2018/05/09 全球购物
麦当劳印度网上订餐:McDelivery
2020/03/16 全球购物
生物制药专业求职信
2014/03/11 职场文书
创业计划书之o2o水果店
2019/08/30 职场文书
Python爬虫之爬取二手房信息
2021/04/27 Python