心形模糊

from mtcnn.mtcnn import MTCNN
import cv2
import numpy as np
img=cv2.imread("C:/Users/xpp/Desktop/Lena.png")#读取图像
detector=MTCNN()
face_list=detector.detect_faces(img)#人脸检测与对齐 
def generate_mask(img_height,img_width,radius,center_x,center_y):
    y,x=np.ogrid[0:img_height,0:img_width]
    scale=5/radius
    mask=5*((-x+center_x)*scale)**2-6*np.abs((-x+center_x)*scale)*((-y+center_y)*scale)+5*((-y+center_y)*scale)**2<128#心形模糊
    return mask
mask_img=np.ones(img.shape,np.int8)
kernel_size=15
blur_img=cv2.blur(img,(kernel_size,kernel_size))
for face in face_list:
    box=face["box"]
    x,y,w,h=box#边界框
    mask=generate_mask(img.shape[0],img.shape[1],max(w,h)/2,x+w/2,y+h/2)
    mask_img[mask]=[0,0,0]
mask_img_verse=np.ones(img.shape,np.int8)-mask_img
result_img=mask_img*img+mask_img_verse*blur_img
cv2.imwrite("C:/Users/xpp/Desktop/result.png",result_img)