github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-21 22:13:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Missing return(image) in draw_rect_landmarks and draw_landmarks

Browse source
This commit is contained in:
charleswg 2021-05-19 23:27:08 -05:00 • committed by GitHub
commit 00b2600393
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
facelib/LandmarksProcessor.py
View file

@ -718,6 +718,7 @@ def draw_landmarks (image, image_landmarks, color=(0,255,0), draw_circles=True,
if transparent_mask: if transparent_mask:
mask = get_image_hull_mask (image.shape, image_landmarks) mask = get_image_hull_mask (image.shape, image_landmarks)
image[...] = ( image * (1-mask) + image * mask / 2 )[...] image[...] = ( image * (1-mask) + image * mask / 2 )[...]
return(image)
def draw_rect_landmarks (image, rect, image_landmarks, face_type, face_size=256, transparent_mask=False, landmarks_color=(0,255,0)): def draw_rect_landmarks (image, rect, image_landmarks, face_type, face_size=256, transparent_mask=False, landmarks_color=(0,255,0)):
draw_landmarks(image, image_landmarks, color=landmarks_color, transparent_mask=transparent_mask) draw_landmarks(image, image_landmarks, color=landmarks_color, transparent_mask=transparent_mask)
@ -729,7 +730,8 @@ def draw_rect_landmarks (image, rect, image_landmarks, face_type, face_size=256,
points = transform_points ( [ ( int(face_size*0.05), 0), ( int(face_size*0.1), int(face_size*0.1) ), ( 0, int(face_size*0.1) ) ], image_to_face_mat, True) points = transform_points ( [ ( int(face_size*0.05), 0), ( int(face_size*0.1), int(face_size*0.1) ), ( 0, int(face_size*0.1) ) ], image_to_face_mat, True)
imagelib.draw_polygon (image, points, (0,0,255), 2) imagelib.draw_polygon (image, points, (0,0,255), 2)
return(image)
def calc_face_pitch(landmarks): def calc_face_pitch(landmarks):
if not isinstance(landmarks, np.ndarray): if not isinstance(landmarks, np.ndarray):
landmarks = np.array (landmarks) landmarks = np.array (landmarks)
@ -899,4 +901,4 @@ def get_transform_mat (image_landmarks, output_size, face_type, scale=1.0):
mat = cv2.getAffineTransform(l_t,pts2) mat = cv2.getAffineTransform(l_t,pts2)
return mat return mat
""" """