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

DockerFile for Mac users to run DeepfaceLab with CPU Mode (#95)

Browse source 
* fix localization nullpointer exception

* fix devicelib error line:61,remove e

* support create docker from cpu dockerfile

* support preview or not when train(resolve cannot connect to X server)
This commit is contained in:
Plucky 2019-01-01 22:08:21 +08:00 • committed by iperov
parent 46bda1d683
commit a8694b73f0
11 changed files with 247 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

9
mainscripts/Trainer.py
View file

@ -277,13 +277,14 @@ def previewThread (input_queue, output_queue):
cv2.destroyAllWindows()
def main (training_data_src_dir, training_data_dst_dir, model_path, model_name, **in_options):
print ("Running trainer.\r\n")
def main (training_data_src_dir, training_data_dst_dir, model_path, model_name,preview, **in_options):
print ("Running trainer(preview=%s).\r\n" % (preview))
output_queue = queue.Queue()
input_queue = queue.Queue()
import threading
thread = threading.Thread(target=trainerThread, args=(output_queue, input_queue, training_data_src_dir, training_data_dst_dir, model_path, model_name), kwargs=in_options )
thread.start()
previewThread (input_queue, output_queue)
if preview:
previewThread (input_queue, output_queue)