使用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 相关文章推荐
使用IronPython把Python脚本集成到.NET程序中的教程
Mar 31 Python
Python实现针对给定单链表删除指定节点的方法
Apr 12 Python
python调用百度地图WEB服务API获取地点对应坐标值
Jan 16 Python
浅谈PYTHON 关于文件的操作
Mar 19 Python
python识别图像并提取文字的实现方法
Jun 28 Python
Django Rest framework三种分页方式详解
Jul 26 Python
Python 变量的创建过程详解
Sep 02 Python
Python使用selenium + headless chrome获取网页内容的方法示例
Oct 16 Python
使用Python制作缩放自如的圣诞老人(圣诞树)
Dec 25 Python
解决Python spyder显示不全df列和行的问题
Apr 20 Python
在spyder IPython console中,运行代码加入参数的实例
Apr 20 Python
python list的index()和find()的实现
Nov 16 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 session 检测和注销
2009/03/16 PHP
用PHP读取和编写XML DOM的实现代码
2011/02/03 PHP
PHP高自定义性安全验证码代码
2011/11/27 PHP
php浏览历史记录的方法
2015/03/10 PHP
php的4种常用运行方式详解
2016/12/22 PHP
关于Laravel Route重定向的一个注意点
2017/01/16 PHP
PHP获取中国时间(上海时区时间)及美国时间的方法
2017/02/23 PHP
javascript事件问题
2009/09/05 Javascript
基于JQUERY的两个ListBox子项互相调整的实现代码
2011/05/07 Javascript
网站繁简切换的JS遇到页面卡死的解决方法
2014/03/12 Javascript
angularjs表格ng-table使用备忘录
2016/03/09 Javascript
js实现旋转木马效果
2017/03/17 Javascript
使用InstantClick.js让页面提前加载200ms
2017/09/12 Javascript
Node.js创建Web、TCP服务器
2017/12/05 Javascript
vue-cli 3.x 配置Axios(proxyTable)跨域代理方法
2018/09/19 Javascript
详解react阻止无效重渲染的多种方式
2018/12/11 Javascript
如何对react hooks进行单元测试的方法
2019/08/14 Javascript
js实现图片区域可点击大小随意改变(适用移动端)代码实例
2019/09/11 Javascript
python执行子进程实现进程间通信的方法
2015/06/02 Python
Python爬虫DOTA排行榜爬取实例(分享)
2017/06/13 Python
Python 私有函数的实例详解
2017/09/11 Python
Python安装模块的常见问题及解决方法
2018/02/05 Python
如何在mac版pycharm选择python版本
2020/07/21 Python
MoviePy简介及Python视频剪辑自动化
2020/12/18 Python
Python使用paramiko连接远程服务器执行Shell命令的实现
2021/03/04 Python
让ie浏览器成为支持html5的浏览器的解决方法(使用html5shiv)
2014/04/08 HTML / CSS
PHP如何与mysql建立链接
2013/05/05 面试题
介绍一下Python中webbrowser的用法
2013/05/07 面试题
工作违纪检讨书
2014/02/17 职场文书
2014年小学植树节活动方案
2014/03/02 职场文书
2015秋季幼儿园开学寄语
2015/03/25 职场文书
品德与社会教学反思
2016/02/24 职场文书
幼儿园2016年圣诞活动总结
2016/03/31 职场文书
用Python实现Newton插值法
2021/04/17 Python
windows server2016安装oracle 11g的图文教程
2022/07/15 Servers
windows10 家庭版下FTP服务器搭建教程
2022/08/05 Servers