From f3782a012bb4aa2e0251fe754e50cf2e67c1ba51 Mon Sep 17 00:00:00 2001 From: iperov Date: Tue, 1 Jan 2019 19:50:31 +0400 Subject: [PATCH] Update Trainer.py --- mainscripts/Trainer.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index b97d7fe..6f7ccef 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -277,7 +277,7 @@ def previewThread (input_queue, output_queue): cv2.destroyAllWindows() -def main (training_data_src_dir, training_data_dst_dir, model_path, model_name, disable_preview, **in_options): +def main (training_data_src_dir, training_data_dst_dir, model_path, model_name, **in_options): print ("Running trainer.\r\n") output_queue = queue.Queue() @@ -286,5 +286,4 @@ def main (training_data_src_dir, training_data_dst_dir, model_path, model_name, 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() - if not disable_preview: - previewThread (input_queue, output_queue) + previewThread (input_queue, output_queue)