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

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

2
facelib/FANExtractor.py
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)