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

1

Browse source
This commit is contained in:
iperov 2018-07-09 00:09:24 +04:00
parent e20527254b
commit a1ff86a6b4
1 changed files with 24 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

13
facelib/LandmarksProcessor.py
View file

@ -81,7 +81,7 @@ def transform_points(points, mat, invert=False):
def get_image_hull_mask (image, image_landmarks): def get_image_hull_mask (image, image_landmarks):
if len(image_landmarks) != 68: if len(image_landmarks) != 68:
raise Exception('get_image_hull_mask work only with 68 landmarks') raise Exception('get_image_hull_mask works only with 68 landmarks')
hull_mask = np.zeros(image.shape[0:2]+(1,),dtype=np.float32) hull_mask = np.zeros(image.shape[0:2]+(1,),dtype=np.float32)
@ -93,7 +93,7 @@ def get_image_hull_mask (image, image_landmarks):
def get_image_eye_mask (image, image_landmarks): def get_image_eye_mask (image, image_landmarks):
if len(image_landmarks) != 68: if len(image_landmarks) != 68:
raise Exception('get_image_eye_mask work only with 68 landmarks') raise Exception('get_image_eye_mask works only with 68 landmarks')
hull_mask = np.zeros(image.shape[0:2]+(1,),dtype=np.float32) hull_mask = np.zeros(image.shape[0:2]+(1,),dtype=np.float32)
@ -174,7 +174,9 @@ def mirror_landmarks (landmarks, val):
return result return result
def draw_landmarks (image, image_landmarks, color): def draw_landmarks (image, image_landmarks, color):
if len(image_landmarks) == 68: if len(image_landmarks) != 68:
raise Exception('get_image_eye_mask works only with 68 landmarks')
jaw = image_landmarks[slice(*landmarks_68_pt["jaw"])] jaw = image_landmarks[slice(*landmarks_68_pt["jaw"])]
right_eyebrow = image_landmarks[slice(*landmarks_68_pt["right_eyebrow"])] right_eyebrow = image_landmarks[slice(*landmarks_68_pt["right_eyebrow"])]
left_eyebrow = image_landmarks[slice(*landmarks_68_pt["left_eyebrow"])] left_eyebrow = image_landmarks[slice(*landmarks_68_pt["left_eyebrow"])]
@ -195,11 +197,6 @@ def draw_landmarks (image, image_landmarks, color):
# jaw big circles # jaw big circles
for x, y in jaw: for x, y in jaw:
cv2.circle(image, (x, y), 2, color, lineType=cv2.LINE_AA) cv2.circle(image, (x, y), 2, color, lineType=cv2.LINE_AA)
else:
for i, (x, y) in enumerate(image_landmarks):
cv2.circle(image, (x, y), 2, color, -1, lineType=cv2.LINE_AA)
#text_color = colorsys.hsv_to_rgb ( (i%4) * (0.25), 1.0, 1.0 )
#cv2.putText(image, str(i), (x, y), cv2.FONT_HERSHEY_SIMPLEX, 0.1,text_color,1)
def draw_rect_landmarks (image, rect, image_landmarks, face_size, face_type): def draw_rect_landmarks (image, rect, image_landmarks, face_size, face_type):
image_utils.draw_rect (image, rect, (255,0,0), 2 ) image_utils.draw_rect (image, rect, (255,0,0), 2 )