云南龙网

  1. 首页 > 云知道 > >

头像识别 头像识别( 二 )

云知道头像识别


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(),

推荐阅读


上一篇：霍比特人是矮人吗

下一篇：蚂蚱产卵是什么样子的