都在发做圣诞树,我来点不一样的,用python给对象戴一顶圣诞帽
既然是要做人脸识别,那么照骗就必须要真人正面照,不然你整个侧面的也识别不了撒~识别不了,那帽子它就会自动变色,越来越绿,飞到了你的头顶…接下来用 dlib 的正脸检测器进行人脸检测,用 dlib 提供的模型提取人脸的五个关键点。
既然是要做人脸识别,那么照骗就必须要真人正面照,不然你整个侧面的也识别不了撒~
识别不了,那帽子它就会自动变色,越来越绿,飞到了你的头顶…
接下来用 dlib 的正脸检测器进行人脸检测,用 dlib 提供的模型提取人脸的五个关键点。
人脸关键点检测器
predictor_path = “shape_predictor_5_face_landmarks.dat”
predictor = dlib.shape_predictor(predictor_path)
正脸检测器
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()
图片打印效果
自动调节帽子大小
选取两个眼角的点,求中心作为放置帽子的x方向的参考坐标,y 方向的坐标用人脸框上线的 y 坐标表示。
根据人脸检测得到的人脸的大小调整帽子的大小,使得帽子大小合适。
所以啊,女朋友照片的脸有多正面做出来的效果就有多好。
选取左右眼眼角的点
point1 = shape.part(0)
point2 = shape.part(2)
求两点中心
eyes_center = ((point1.x+point2.x)//2,(point1.y+point2.y)//2)
根据人脸大小调整帽子大小
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 = 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’)
提取后效果图
再提取帽子区域
hat = cv2.bitwise_and(resized_hat,resized_hat,mask = mask)
效果图
给女朋友戴帽子
帽子处理好了就该给她戴上去了,把提取的圣诞帽区域和图片中提取的区域相加后放到原图中去。
注意:相加之前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
最后的效果图
================================================================
效果展示
女朋友太多了,这里就选用现女友了,免得这些前女友又回来找我。
GUI界面实现代码
import PySimpleGUI as sg
import os.path
import cv2
file_list_column = [
[sg.Submit(‘立即戴帽’, key=‘Go’, size=(15, 1)), sg.Cancel(‘溜了溜了’, key=‘Cancel’, size=(15, 1))],
[
sg.Text(“图片存放位置”),
sg.In(size=(25, 1), enable_events=True, key=“-FOLDER-”),
sg.FolderBrowse(‘选择’),
],
[
sg.Listbox(
values=[], enable_events=True, size=(40, 20), key=“-FILE LIST-”
)
]
]
image_viewer_column = [
[sg.Text(“从左边女朋友中选择一个:”)],
[sg.Image(key=“-IMAGE-”)]
]
layout = [
[
sg.Column(file_list_column),
sg.VSeperator(),
sg.Column(image_viewer_column),
]
]
window = sg.Window(“给女朋友添加一个圣诞帽”, layout)
filename = ‘’
while True:
event, values = window.read()
if event == “Cancel” or event == sg.WIN_CLOSED:
break
if event == “-FOLDER-”:
folder = values[“-FOLDER-”]
try:
file_list = os.listdir(folder)
except:
file_list = []
fnames = [
f
for f in file_list
if os.path.isfile(os.path.join(folder, f))
and f.lower().endswith((“.jpg”, “.png”))
]
window[“-FILE LIST-”].update(fnames)
elif event == “-FILE LIST-”:
try:
filename = os.path.join(values[“-FOLDER-”], values[“-FILE LIST-”][0])
if filename.endswith(‘.jpg’):
im = cv2.imread(filename)
cv2.imwrite(filename.replace(‘jpg’, ‘png’), im)
window[“-IMAGE-”].update(filename=filename.replace(‘jpg’, ‘png’))
except Exception as e:
print(e)
elif event == “Go”:
try:
output = add_hat(filename)
展示效果
cv2.imshow(“output”,output)
cv2.waitKey(0)
cv2.imwrite(“output.png”,output)
print(output)
window[“-IMAGE-”].update(filename=‘output.png’)
except:
print(‘OMG!添加失败了!’)
cv2.destroyAllWindows()
界面是用 PySimpleGUI 框架做的,分为左右两部分,左边是确定/取消/文件夹选择/图片文件列表,右边是图片展示栏。
选择好存放照片的文件夹后,你准备好的女朋友照片会在图片文件列表显示出来,这里图片是没限制的,JPG/PNG格式的都可以用。
选择列表里的女朋友点一下,就可以在右边显示出来照片。
你选中哪张,后面就给谁带上了帽子…
那么为什么这里就不限制图片格式了呢?
因为 PySimpleGUI 的图片展示默认只支持 png 格式的,这部分我加了一个判断,如果是 jpg 格式的图片,就用 cv2 转换成 png 格式再进行展示。
关键步骤到这里已经完成了,然后将两部分代码整合。
在 GUI 界面上点一下 “立即戴帽” 按钮时,后台接收到图片的路径,传递给我们的图片处理函数,在处理完后将图片保存在文件夹下,并更新在 GUI 界面右边的展示的图片即可。
最终效果展示
====================================================================
依然是用常用的 pyinstaller 打包,模块安装我就不讲了吧,前面装了好几个了,再不会拉出去关禁闭~
那我说快点
学好 Python 不论是就业还是做副业赚钱都不错,但要学会 Python 还是要有一个学习规划。最后大家分享一份全套的 Python 学习资料,给那些想学习 Python 的小伙伴们一点帮助!
一、Python所有方向的学习路线
Python所有方向路线就是把Python常用的技术点做整理,形成各个领域的知识点汇总,它的用处就在于,你可以按照上面的知识点去找对应的学习资源,保证自己学得较为全面。
二、学习软件
工欲善其事必先利其器。学习Python常用的开发软件都在这里了,给大家节省了很多时间。
三、全套PDF电子书
书籍的好处就在于权威和体系健全,刚开始学习的时候你可以只看视频或者听某个人讲课,但等你学完之后,你觉得你掌握了,这时候建议还是得去看一下书籍,看权威技术书籍也是每个程序员必经之路。
四、入门学习视频
我们在看视频学习的时候,不能光动眼动脑不动手,比较科学的学习方法是在理解之后运用它们,这时候练手项目就很适合了。
五、实战案例
光学理论是没用的,要学会跟着一起敲,要动手实操,才能将自己的所学运用到实际当中去,这时候可以搞点实战案例来学习。
六、面试资料
我们学习Python必然是为了找到高薪的工作,下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料,并且有阿里大佬给出了权威的解答,刷完这一套面试资料相信大家都能找到满意的工作。
网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。
一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。
更多推荐
所有评论(0)