david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix FANExtractor

Browse Source
This commit is contained in:
Colombo
2020-02-20 12:20:18 +04:00
parent e6e11ca056
commit 090d4a5be3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facelib/FANExtractor.py
+1 -1
View File
@@ -216,7 +216,7 @@ class FANExtractor(object):
for i, lmrks in enumerate(landmarks): for i, lmrks in enumerate(landmarks):
try: try:
if lmrks is not None: if lmrks is not None:
image_to_face_mat = LandmarksProcessor.get_transform_mat (lmrks, 256, FaceType.FULL, full_face_align_top=False) image_to_face_mat = LandmarksProcessor.get_transform_mat (lmrks, 256, FaceType.FULL)
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(face_image, is_bgr=is_bgr) rects2 = second_pass_extractor.extract(face_image, is_bgr=is_bgr)