def get_mouth_mask(): mouth_mask = LandmarksProcessor.get_image_mouth_mask( sample_bgr.shape, sample_landmarks) # set eye masks to 2-3 clip = np.clip(mouth_mask, 0, 1) clip[clip > 0.1] += 2 return clip
def get_eyes_mouth_mask(): eyes_mask = LandmarksProcessor.get_image_eye_mask( sample_bgr.shape, sample_landmarks) mouth_mask = LandmarksProcessor.get_image_mouth_mask( sample_bgr.shape, sample_landmarks) mask = eyes_mask + mouth_mask return np.clip(mask, 0, 1)