68关键点------人脸掩膜mask

import os
import cv2
import dlib
import numpy as np
from imutils import face_utils
"""
大论文用到的代码
代码可视化
人脸对齐3
"""


def face_remap(shape):
    shape = face_utils.shape_to_np(shape)
    remapped_image = shape.copy()
    # left eye brow
    remapped_image[17] = shape[26]
    remapped_image[18] = shape[25]
    remapped_image[19] = shape[24]
    remapped_image[20] = shape[23]
    remapped_image[21] = shape[22]
    # right eye brow
    remapped_image[22] = shape[21]
    remapped_image[23] = shape[20]
    remapped_image[24] = shape[19]
    remapped_image[25] = shape[18]
    remapped_image[26] = shape[17]
    # neatening
    remapped_image[27] = shape[0]

    remapped_image = cv2.convexHull(shape)  # 输出凸包结果，n * 1 *2 数据结构，n为外包围圈点数
    return remapped_image

frame = cv2.imread('family.png')

detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor("./pth/shape_predictor_68_face_landmarks.dat")

dets = detector(frame, 1)
mask = np.zeros((frame.shape[0], frame.shape[1]))  # mask与face一般大小

for det in dets:
    shape = predictor(frame, det)

    rshape =face_remap(shape)
    cv2.fillConvexPoly(mask, rshape[0:27], 1)  # 多边形填充 1为白色 [0:27]为外围点


out = np.zeros_like(frame)
cv2.imshow("mask", mask)  # out皮肤区域最大化 用于计算的最好ROI
mask = mask.astype(np.bool)
out[mask] = frame[mask]  # out 除了脸部其他都是黑色的
cv2.imshow("out", out)  # out皮肤区域最大化 用于计算的最好ROI

cv2.imshow('frame', frame)
cv2.waitKey()