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

Extractor: 2nd pass now faster on frames where faces are not found

Browse source
This commit is contained in:
Colombo 2019-12-22 10:46:44 +04:00
parent 8305832a27
commit 9312a3c334
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
mainscripts/Extractor.py
View file

@ -110,8 +110,11 @@ class ExtractSubprocessor(Subprocessor):
#override
def process_data(self, data):
filename_path = Path( data.filename )
filename_path_str = str(filename_path)
if self.type == 'landmarks' and len(data.rects) == 0:
return data
if self.cached_image[0] == filename_path_str:
image = self.cached_image[1] #cached image for manual extractor
else:
@ -163,8 +166,7 @@ class ExtractSubprocessor(Subprocessor):
return data
elif self.type == 'landmarks':
elif self.type == 'landmarks':
if data.rects_rotation == 0:
rotated_image = image
elif data.rects_rotation == 90: