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 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

4
mainscripts/Extractor.py
View file

@ -151,7 +151,9 @@ class ExtractSubprocessor(Subprocessor):
if len(rects) != 0:
data.rects_rotation = rot
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]
return data