使用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中MySQLdb模块用法实例
Nov 10 Python
python中Flask框架简单入门实例
Mar 21 Python
python输出100以内的质数与合数实例代码
Jul 08 Python
Python实现的特征提取操作示例
Dec 03 Python
python 搭建简单的http server,可直接post文件的实例
Jan 03 Python
Python零基础入门学习之输入与输出
Apr 03 Python
Python Opencv实现图像轮廓识别功能
Mar 23 Python
Python爬虫动态ip代理防止被封的方法
Jul 07 Python
python3.8与pyinstaller冲突问题的快速解决方法
Jan 16 Python
Python3搭建http服务器的实现代码
Feb 11 Python
Python实现验证码识别
Jun 15 Python
Django怎么在admin后台注册数据库表
Nov 14 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中exec函数和shell_exec函数的区别
2014/08/20 PHP
asp批量修改记录的代码
2008/06/25 Javascript
jQuery 源码分析笔记(3) Deferred机制
2011/06/19 Javascript
jQuery EasyUI API 中文文档 - Parser 解析器
2011/09/29 Javascript
jquery不会自动回收xmlHttpRequest对象 导致了内存溢出
2012/06/18 Javascript
jquery动态加载select下拉框示例代码
2013/12/10 Javascript
js清理Word格式示例代码
2014/02/13 Javascript
Angularjs 自定义服务的三种方式(推荐)
2016/08/02 Javascript
预防网页挂马的方法总结
2016/11/03 Javascript
JS实现复制功能
2017/03/01 Javascript
利用Angular+Angular-Ui实现分页(代码加简单)
2017/03/10 Javascript
Bootstrap表格制作代码
2017/03/17 Javascript
微信小程序开发之toast等弹框提示使用教程
2017/06/08 Javascript
关于Vue实现组件信息的缓存问题
2017/08/23 Javascript
JS设计模式之单例模式(一)
2017/09/29 Javascript
vue图片加载失败时用默认图片替换的方法
2019/08/29 Javascript
JS实现密码框效果
2020/09/10 Javascript
[01:15:16]DOTA2-DPC中国联赛 正赛 Elephant vs Aster BO3 第一场 1月26日
2021/03/11 DOTA
Python日期操作学习笔记
2008/10/07 Python
python实现linux服务器批量修改密码并生成execl
2014/04/22 Python
python执行等待程序直到第二天零点的方法
2015/04/23 Python
通过Python来使用七牛云存储的方法详解
2015/08/07 Python
浅谈django rest jwt vue 跨域问题
2018/10/26 Python
Python实现查找数组中任意第k大的数字算法示例
2019/01/23 Python
python类的实例化问题解决
2019/08/31 Python
pytorch:实现简单的GAN示例(MNIST数据集)
2020/01/10 Python
简单的HTML5初步入门教程
2015/09/29 HTML / CSS
快时尚眼镜品牌,全国连锁眼镜店:LOHO眼镜生活
2018/10/08 全球购物
什么是java序列化,如何实现java序列化
2012/11/14 面试题
新入职员工的自我介绍演讲稿
2014/01/02 职场文书
仓库主管岗位职责
2014/03/02 职场文书
表决心的诗句大全
2014/03/11 职场文书
产品陈列协议书(标准版)
2014/09/17 职场文书
2014年乡镇工作总结
2014/11/21 职场文书
伊索寓言读书笔记
2015/06/30 职场文书
利用Python网络爬虫爬取各大音乐评论的代码
2021/04/13 Python