看我用Python来给女朋友戴口罩!等等,女朋友呢?
前言
2019 年底开始蔓延的新型肺炎疫情牵动人心,作为个体,我们力所能及的就是尽量待在家中少出门。
看到一些朋友叫设计同学帮忙给自己的头像戴上口罩,作为技术人,心想一定还有更多人有这样的诉求,不如开发一个简单的程序来实现这个需求,也算是帮助设计姐姐减少工作量。
于是花了些时间,写了一个叫做 face-mask[1] 的命令行工具,能够轻松的给图片中的人像戴上口罩,而且口罩的方向和大小都是适应人脸的哦~
使用
安装 face-mask
确保 python 版本在 3.6 及以上
pip install face-mask
使用 face-mask
直接指定图片路径即可为图片中的人像戴上口罩,并会生成一个新的图片(额外有 -with-mask
后缀):
face-mask /path/to/face/picture
通过指定 --show
选项,还可以使用默认图片查看器打开新生成的图片:
face-mask /path/to/face/picture --show
效果
给一个人戴上口罩
给多个人戴上口罩
给动漫人物戴上口罩
实现
思路
要想实现上面的效果,我们应该怎么做?不妨这么想:
-
首先是识别出人的鼻子(nose_bridge)和脸轮廓(chin)
-
通过脸轮廓确定出脸左点(chin_left_point)、脸底点(chin_bottom_point)和脸右点(chin_right_point)
-
由鼻子和脸底点确定口罩大小的高度、中心线
-
将口罩左右平均分为两个部分
-
调整左口罩大小,宽度为脸左点到中心线的距离
-
调整右口罩大小,宽度为脸右点到中心线的距离
-
合并左右口罩为新口罩
-
旋转新口罩,角度为中心线相对于 y 轴的旋转角
-
将新口罩放在原图适当位置
关于人脸识别,可以使用 face_recognition[2] 库进行识别。
关于图像处理,可以使用 pillow[3] 库进行处理。
代码
有了思路之后,实现就是件相对轻松的事情。不过对库的熟悉和图片的变换计算可能要花些时间。
详细的代码请阅读 face-mask[4]。这里仅说明下最核心的步骤。
人脸识别
借助 face_recognition
库可以轻松的识别出人像,最终得到的 face_landmarks
是一个列表,里面的每个 face_landmark
都表示一个人像数据。
face_landmark
是一个字典,其中的键表示人像特征,值表示该特征的点的列表。比如:
- 键
nose_bridge
表示鼻梁 - 键
chin
表示脸颊
我们需要根据每个 face_landmark
,给对应的头像戴上口罩。
获得鼻子和脸颊的特征点
通过上述代码,我们获得了:
- 表示上鼻梁的一个点
nose_point
- 表示脸左点
chin_left_point
- 表示脸右点
chin_right_point
- 表示脸底点
chin_bottom_point
拆分、缩放和合并口罩
上述代码主要做了如下内容:
- 将口罩左右平均分为两个部分
- 调整左口罩大小,宽度为脸左点到中心线的距离 * 宽度系数 1.2
- 调整右口罩大小,宽度为脸右点到中心线的距离 * 宽度系数 1.2
- 合并左右口罩为新口罩
get_distance_from_point_to_line
用来获取一个点到一条线的距离,具体实现可看源代码。
width_ratio
是宽度系数,用来适当扩大口罩。原因我们是根据脸颊的宽度计算口罩的宽度,但口罩是待在耳朵上的,真实宽度应该要更宽。
旋转口罩、并放到原图适当位置
上述代码主要做了如下内容:
- 旋转新口罩,角度为中心线相对于 y 轴的旋转角
- 计算口罩应该放置的坐标
- 将新口罩放在原图的计算出的坐标下
最后就是将新图片保存到本地路径,代码不再展示。
总结
我们借助 face_recognition
库可以轻松的识别出人像,然后根据脸颊的宽度和鼻梁位置计算出口罩的大小、方向和位置,并最终生成出戴上口罩的图片。整个过程并不复杂,但在坐标计算上要格外小心,如此,我们便打造了一个短小精悍的“自动戴上口罩”程序!很多小伙伴在学习python的过程中往往因为没有资料或者没人指导从而导致自己不想学下去了,因此我特意准备了大量的pdf书籍、视频教程,都免费送给大家!不管你是零基础还是有基础都可以获取到自己相对应的学习礼包!加群592539176,包括python软件工具和2019最新入门到实战教程!