人脸图片加口罩

实现对图片加口罩操作

from PIL import Image
import dlib
import numpy as np
import cv2
from PIL import Image


def christmas(img,x,y,w,h):
    im = Image.fromarray(cv2.cvtColor(img,cv2.COLOR_BGR2RGB))
    mark=Image.open("口.png")
    mark = mark.resize((w, h))
    layer=Image.new('RGBA', im.size, (0,0,0,0))
    layer.paste(mark, (x,y))
    out=Image.composite(layer,im,layer)
    img = cv2.cvtColor(np.asarray(out),cv2.COLOR_RGB2BGR)
    return img


predictor_path = 'shape_predictor_68_face_landmarks.dat'
detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor(predictor_path)

frame = cv2.imread('img\\3.png')
dets = detector(frame, 0)
for k, d in enumerate(dets):
    x = []
    y = []
    height = d.bottom() - d.top()
    width = d.right() - d.left()
    shape = predictor(frame, d)
    for i in range(48, 68):
        x.append(shape.part(i).x)
        y.append(shape.part(i).y)
    y_max = (int)(max(y) + height / 4)
    y_min = (int)(min(y) - height / 4)
    x_max = (int)(max(x) + width / 4)
    x_min = (int)(min(x) - width / 4)
frame=christmas(frame,x_min,y_min,x_max - x_min,y_max - y_min)
cv2.imshow('frame', frame)
cv2.waitKey(0)