CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

extractor: increased speed of 1st pass by DLIB for >= 4GB vram cards,

Browse Source 
added 'h' key for hide the help in manual extractor
This commit is contained in:
iperov
2019-03-08 17:01:11 +04:00
parent bced4d3259
commit 168d48b119
2 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
joblib/SubprocessorBase.py
+5 -2
View File
@@ -48,7 +48,7 @@ class Subprocessor(object):
return "undefined"
def log_info(self, msg): self.c2s.put ( {'op': 'log_info', 'msg':msg } )
def log_err(self, msg): self.c2s.put ( {'op': 'log_err' , 'msg':msg } )
def log_err(self, msg): self.c2s.put ( {'op': 'log_err' , 'msg':msg } )
def progress_bar_inc(self, c): self.c2s.put ( {'op': 'progress_bar_inc' , 'c':c } )
def _subprocess_run(self, client_dict):
@@ -163,6 +163,10 @@ class Subprocessor(object):
op = obj.get('op','')
if op == 'init_ok':
cli.state = 0
elif op == 'log_info':
io.log_info(obj['msg'])
elif op == 'log_err':
io.log_err(obj['msg'])
elif op == 'error':
cli.kill()
self.clis.remove(cli)
@@ -184,7 +188,6 @@ class Subprocessor(object):
while not cli.c2s.empty():
obj = cli.c2s.get()
op = obj.get('op','')
if op == 'success':
#success processed data, return data and result to on_result
self.on_result (cli.host_dict, obj['data'], obj['result'])