github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-16 10:03:41 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Better face landmark representation

Browse source 
- fixed nose missing one point
- made AA lines instead of regular ones
- made closed polylines for eyes and mouth
- reduced radius of circles for eyes, nose, and mouth as they obscure lines for small images
This commit is contained in:
Artem Ivanov 2018-07-07 04:59:17 +03:00 • committed by GitHub
parent bda1eed582
commit 48e281f675
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 47 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

58
facelib/LandmarksProcessor.py
View file

@ -29,12 +29,12 @@ landmarks_2D = np.stack( [ mean_face_x, mean_face_y ], axis=1 )
# 68 point landmark definitions
landmarks_68_pt = { "mouth": (48,68),
"right_eyebrow": (17,22),
"left_eyebrow": (22,27),
"right_eye": (36,42),
"left_eye": (42,48),
"nose": (27,35),
"jaw": (0,17) }
"right_eyebrow": (17, 22),
"left_eyebrow": (22, 27),
"right_eye": (36, 42),
"left_eye": (42, 48),
"nose": (27, 36), # missed one point
"jaw": (0, 17) }
def get_transform_mat (image_landmarks, output_size, face_type):
if not isinstance(image_landmarks, np.ndarray):
@ -174,17 +174,47 @@ def mirror_landmarks (landmarks, val):
return result
def draw_landmarks (image, image_landmarks, color):
if len(image_landmarks) == 68:
jaw = [image_landmarks[pt_idx] for pt_idx in range(*landmarks_68_pt["jaw"])]
for x, y in jaw:
cv2.circle(image, (x, y), 2, color, lineType=cv2.LINE_AA)
cv2.polylines(image, np.array([jaw]), False, color, lineType=cv2.LINE_AA)
right_eyebrow = [image_landmarks[pt_idx] for pt_idx in range(*landmarks_68_pt["right_eyebrow"])]
cv2.polylines(image, np.array([right_eyebrow]), False, color, lineType=cv2.LINE_AA)
for x, y in right_eyebrow:
cv2.circle(image, (x, y), 1, color, 1, lineType=cv2.LINE_AA)
left_eyebrow = [image_landmarks[pt_idx] for pt_idx in range(*landmarks_68_pt["left_eyebrow"])]
cv2.polylines(image, np.array([left_eyebrow]), False, color, lineType=cv2.LINE_AA)
for x, y in left_eyebrow:
cv2.circle(image, (x, y), 1, color, 1, lineType=cv2.LINE_AA)
mouth = [image_landmarks[pt_idx] for pt_idx in range(*landmarks_68_pt["mouth"])]
cv2.polylines(image, np.array([mouth]), True, color, lineType=cv2.LINE_AA)
for x, y in mouth:
cv2.circle(image, (x, y), 1, color, 1, lineType=cv2.LINE_AA)
right_eye = [image_landmarks[pt_idx] for pt_idx in range(*landmarks_68_pt["right_eye"])]
cv2.polylines(image, np.array([right_eye]), True, color, lineType=cv2.LINE_AA)
for x, y in right_eye:
cv2.circle(image, (x, y), 1, color, 1, lineType=cv2.LINE_AA)
left_eye = [image_landmarks[pt_idx] for pt_idx in range(*landmarks_68_pt["left_eye"])]
cv2.polylines(image, np.array([left_eye]), True, color, lineType=cv2.LINE_AA)
for x, y in left_eye:
cv2.circle(image, (x, y), 1, color, 1, lineType=cv2.LINE_AA)
nose = [image_landmarks[pt_idx] for pt_idx in range(*landmarks_68_pt["nose"])]
cv2.polylines(image, np.array([nose+[nose[-6]]]), False, color, lineType=cv2.LINE_AA)
for x, y in nose:
cv2.circle(image, (x, y), 1, color, 1, lineType=cv2.LINE_AA)
else:
for i, (x, y) in enumerate(image_landmarks):
cv2.circle(image, (x, y), 2, color, -1)
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)
if len(image_landmarks) == 68:
for feat,idx_range in landmarks_68_pt.items():
for idx in range( idx_range[0], idx_range[1] - 1 ):
pt1 = tuple(image_landmarks[idx])
pt2 = tuple(image_landmarks[idx + 1])
cv2.line( image, pt1, pt2, color )
def draw_rect_landmarks (image, rect, image_landmarks, face_size, face_type):
image_utils.draw_rect (image, rect, (255,0,0), 2 )