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

fix extractor

Browse Source
This commit is contained in:
Colombo
2020-07-03 16:06:06 +04:00
parent 4ce4997d1a
commit 57ea3e61b7
1 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mainscripts/Extractor.py
+3 -1
View File
@@ -151,7 +151,9 @@ class ExtractSubprocessor(Subprocessor):
if len(rects) != 0: if len(rects) != 0:
data.rects_rotation = rot data.rects_rotation = rot
break break
if max_faces_from_image is not None and len(data.rects) > 1: if max_faces_from_image is not None and \
max_faces_from_image > 0 and \
len(data.rects) > 0:
data.rects = data.rects[0:max_faces_from_image] data.rects = data.rects[0:max_faces_from_image]
return data return data