diff --git a/facelib/LandmarksProcessor.py b/facelib/LandmarksProcessor.py index 8eed2d0..88470ff 100644 --- a/facelib/LandmarksProcessor.py +++ b/facelib/LandmarksProcessor.py @@ -310,8 +310,8 @@ def draw_landmarks (image, image_landmarks, color=(0,255,0), transparent_mask=Fa mask = get_image_hull_mask (image.shape, image_landmarks) image[...] = ( image * (1-mask) + image * mask / 2 )[...] -def draw_rect_landmarks (image, rect, image_landmarks, face_size, face_type, transparent_mask=False): - draw_landmarks(image, image_landmarks, transparent_mask=transparent_mask) +def draw_rect_landmarks (image, rect, image_landmarks, face_size, face_type, transparent_mask=False, landmarks_color=(0,255,0) ): + draw_landmarks(image, image_landmarks, color=landmarks_color, transparent_mask=transparent_mask) image_utils.draw_rect (image, rect, (255,0,0), 2 ) image_to_face_mat = get_transform_mat (image_landmarks, face_size, face_type)