From 5c0c79e528390d8fbe49c3871ad4aa13625ac52b Mon Sep 17 00:00:00 2001 From: iperov Date: Tue, 1 Jan 2019 19:42:19 +0400 Subject: [PATCH] fix last PR --- mainscripts/Trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index 0245eb3..b97d7fe 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, preview, **in_options): +def main (training_data_src_dir, training_data_dst_dir, model_path, model_name, disable_preview, **in_options): print ("Running trainer.\r\n") output_queue = queue.Queue() @@ -286,5 +286,5 @@ 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 preview: + if not disable_preview: previewThread (input_queue, output_queue)