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使用Flask框架获取用户IP地址的方法
Mar 21 Python
分享一下如何编写高效且优雅的 Python 代码
Sep 07 Python
python实现决策树分类算法
Dec 21 Python
Python简单实现的代理服务器端口映射功能示例
Apr 08 Python
Python 利用pydub库操作音频文件的方法
Jan 09 Python
python3的UnicodeDecodeError解决方法
Dec 20 Python
python网络编程socket实现服务端、客户端操作详解
Mar 24 Python
python matplotlib实现将图例放在图外
Apr 17 Python
pandas DataFrame运算的实现
Jun 14 Python
浅谈Python里面None True False之间的区别
Jul 09 Python
python如何支持并发方法详解
Jul 25 Python
Python自定义sorted排序实现方法详解
Sep 18 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
windows下PHP APACHE MYSQ完整配置
2007/01/02 PHP
php使用filter过滤器验证邮箱 ipv6地址 url验证
2013/12/25 PHP
基于Laravel-admin 后台的自定义页面用法详解
2019/09/30 PHP
基于JQuery的cookie插件
2010/04/07 Javascript
JQuery 应用 JQuery.groupTable.js
2010/12/15 Javascript
jQuery使用post方法提交数据实例
2015/03/25 Javascript
JavaScript中操作字符串小结
2015/05/04 Javascript
JS实现仿QQ面板的手风琴效果折叠菜单代码
2015/09/11 Javascript
canvas绘制一个常用的emoji表情
2017/03/30 Javascript
jquery仿微信聊天界面
2017/05/06 jQuery
vue 过滤器filter实例详解
2018/03/14 Javascript
详解Angular路由之路由守卫
2018/05/10 Javascript
详解Vue+Element的动态表单,动态表格(后端发送配置,前端动态生成)
2019/04/20 Javascript
p5.js临摹动态图形的方法
2019/10/23 Javascript
解决 window.onload 被覆盖的问题方法
2020/01/14 Javascript
JavaScript enum枚举类型定义及使用方法
2020/05/15 Javascript
[01:15]《辉夜杯》北京网鱼队巡礼
2015/10/26 DOTA
python学习数据结构实例代码
2015/05/11 Python
Python的Django框架中的数据过滤功能
2015/07/17 Python
Python多线程threading和multiprocessing模块实例解析
2018/01/29 Python
Django项目中包含多个应用时对url的配置方法
2018/05/30 Python
详解python中Numpy的属性与创建矩阵
2018/09/10 Python
使用Python进行体育竞技分析(预测球队成绩)
2019/05/16 Python
python写日志文件操作类与应用示例
2019/07/01 Python
python学习笔记之多进程
2020/08/06 Python
python爬虫爬取网页数据并解析数据
2020/09/18 Python
米兰网婚纱礼服法国网上商店:Milanoo法国
2016/08/20 全球购物
Shopee越南:东南亚与台湾电商平台
2019/02/03 全球购物
九年级历史教学反思
2014/01/27 职场文书
人力资源经理的岗位职责
2014/03/02 职场文书
民族团结先进集体事迹材料
2014/05/22 职场文书
师范生求职自荐信
2014/06/14 职场文书
2014年学校领导班子对照检查材料
2014/09/19 职场文书
《兰兰过桥》教学反思
2016/02/20 职场文书
python机器学习创建基于规则聊天机器人过程示例详解
2021/11/02 Python
HTML5之高度塌陷问题的解决
2022/06/01 HTML / CSS