diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index 2c6ae83..0245eb3 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -277,8 +277,8 @@ 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): - print ("Running trainer(preview=%s).\r\n" % (preview)) +def main (training_data_src_dir, training_data_dst_dir, model_path, model_name, preview, **in_options): + print ("Running trainer.\r\n") output_queue = queue.Queue() input_queue = queue.Queue() @@ -287,4 +287,4 @@ def main (training_data_src_dir, training_data_dst_dir, model_path, model_name,p thread.start() if preview: - previewThread (input_queue, output_queue) \ No newline at end of file + previewThread (input_queue, output_queue)