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

fix

Browse source
This commit is contained in:
iperov 2019-03-18 10:38:48 +04:00
parent b8efb4cbba
commit da7cf225db
1 changed files with 18 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

6
facelib/LandmarksExtractor.py
View file

@ -44,11 +44,11 @@ class LandmarksExtractor(object):
pts_img = [ ( int(pt[0]), int(pt[1]) ) for pt in pts_img ] pts_img = [ ( int(pt[0]), int(pt[1]) ) for pt in pts_img ]
landmarks.append ( ( (left, top, right, bottom),pts_img ) ) landmarks.append ( ( (left, top, right, bottom),pts_img ) )
except Exception as e: except Exception as e:
print ("extract_from_bgr: ", traceback.format_exc() )
landmarks.append ( ( (left, top, right, bottom), None ) ) landmarks.append ( ( (left, top, right, bottom), None ) )
if second_pass_extractor is not None: if second_pass_extractor is not None:
for i in range(len(landmarks)): for i in range(len(landmarks)):
try:
rect, lmrks = landmarks[i] rect, lmrks = landmarks[i]
if lmrks is None: if lmrks is None:
continue continue
@ -57,7 +57,7 @@ class LandmarksExtractor(object):
face_image = cv2.warpAffine(input_image, image_to_face_mat, (256, 256), cv2.INTER_CUBIC) face_image = cv2.warpAffine(input_image, image_to_face_mat, (256, 256), cv2.INTER_CUBIC)
rects2 = second_pass_extractor.extract_from_bgr(face_image) rects2 = second_pass_extractor.extract_from_bgr(face_image)
if len(rects2) != 1: #dont do second pass if more than 1 face detected in cropped image if len(rects2) != 1: #dont do second pass if more than 1 or zero faces detected in cropped image
continue continue
rect2 = rects2[0] rect2 = rects2[0]
@ -65,6 +65,8 @@ class LandmarksExtractor(object):
lmrks2 = self.extract_from_bgr (face_image, [rect2] )[0][1] lmrks2 = self.extract_from_bgr (face_image, [rect2] )[0][1]
source_lmrks2 = LandmarksProcessor.transform_points (lmrks2, image_to_face_mat, True) source_lmrks2 = LandmarksProcessor.transform_points (lmrks2, image_to_face_mat, True)
landmarks[i] = (rect, source_lmrks2) landmarks[i] = (rect, source_lmrks2)
except:
continue
return landmarks return landmarks