AI换发型
生活中看到别人的漂亮发型总有一点想尝试的冲动,但却没有一个值得信赖的Tony老师,今天我们使用PaddleHub进行AI换发型。
import cv2
import paddlehub as hub
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np
import math
from matplotlib.path import Path
1.首先得有一张照片(以下照片均来自网络,侵删,仅供学习使用),假如这是你,首先通过PaddleHub改变获取你的脸部关键点。
src_img = cv2.imread('./fce3ef51303eb0eac3a9eb6e76e96d74.jpeg')
module = hub.Module(name="face_landmark_localization")
result = module.keypoint_detection(images=[src_img])
tmp_img = src_img.copy()
for index, point in enumerate(result[0]['data'][0]):
# cv2.putText(img, str(index), (int(point[0]), int(point[1])), cv2.FONT_HERSHEY_COMPLEX, 3, (0,0,255), -1)
cv2.circle(tmp_img, (int(point[0]), int(point[1])), 2, (0, 0, 255), -1)
res_img_path = 'face_landmark.jpg'
cv2.imwrite(res_img_path, tmp_img)
img0 = mpimg.imread(res_img_path)
# 展示预测68个关键点结果
plt.figure(figsize=(5,5))
plt.