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

2
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:
mask = get_image_hull_mask (image.shape, image_landmarks)
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)):
draw_landmarks(image, image_landmarks, color=landmarks_color, transparent_mask=transparent_mask)
@ -729,6 +730,7 @@ 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)
imagelib.draw_polygon (image, points, (0,0,255), 2)
return(image)
def calc_face_pitch(landmarks):
if not isinstance(landmarks, np.ndarray):