AI换脸特惠

import cv2
import dlib

# 加载检测器和预测器
detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")

# 读取源图像和目标图像
source_img = cv2.imread('source.jpg')
target_img = cv2.imread('target.jpg')

# 检测人脸并获取关键点
def get_landmarks(image):
    faces = detector(image)
    if len(faces) > 0:
        return np.matrix([[p.x, p.y] for p in predictor(image, faces[0]).parts()])
    return None

source_landmarks = get_landmarks(source_img)
target_landmarks = get_landmarks(target_img)

# 使用仿射变换进行换脸
M = cv2.getAffineTransform(np.float32(source_landmarks[:3]), np.float32(target_landmarks[:3]))
output = cv2.warpAffine(source_img, M, (target_img.shape[1], target_img.shape[0]))

# 显示结果
cv2.imshow('Output', output)
cv2.waitKey(0)
cv2.destroyAllWindows()