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

increased speed of sort by hist sim for ten thousands of faces

Browse source
This commit is contained in:
iperov 2019-01-04 22:54:09 +04:00
parent 08550ac856
commit b5ba7d52cb
6 changed files with 226 additions and 140 deletions
Download patch file Download diff file Expand all files Collapse all files

8
mainscripts/Converter.py
View file

@ -104,13 +104,13 @@ class ConvertSubprocessor(SubprocessorBase):
return len (self.input_data)
#override
def onHostGetData(self):
def onHostGetData(self, host_dict):
if len (self.input_data) > 0:
return self.input_data.pop(0)
return None
#override
def onHostDataReturn (self, data):
def onHostDataReturn (self, host_dict, data):
self.input_data.insert(0, data)
#overridable
@ -195,13 +195,13 @@ class ConvertSubprocessor(SubprocessorBase):
return (files_processed, faces_processed)
#override
def onHostResult (self, data, result):
def onHostResult (self, host_dict, data, result):
self.files_processed += result[0]
self.faces_processed += result[1]
return 1
#override
def get_start_return(self):
def onFinalizeAndGetResult(self):
return self.files_processed, self.faces_processed
def main (input_dir, output_dir, model_dir, model_name, aligned_dir=None, **in_options):