From 3e175ce1d395396b27638967c9ba8bfd39f86585 Mon Sep 17 00:00:00 2001 From: Luke Barr Date: Mon, 18 Feb 2019 10:36:33 -0600 Subject: [PATCH] Add disable preview option with --disable-preview --- main.py | 2 ++ mainscripts/Trainer.py | 3 ++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index eee0e7f..597dd73 100644 --- a/main.py +++ b/main.py @@ -87,6 +87,7 @@ if __name__ == "__main__": model_path=arguments.model_dir, model_name=arguments.model_name, debug = arguments.debug, + disable_preview = arguments.disable_preview, #**options force_gpu_idx = arguments.force_gpu_idx, cpu_only = arguments.cpu_only @@ -98,6 +99,7 @@ if __name__ == "__main__": train_parser.add_argument('--model-dir', required=True, action=fixPathAction, dest="model_dir", help="Model dir.") train_parser.add_argument('--model', required=True, dest="model_name", choices=Path_utils.get_all_dir_names_startswith ( Path(__file__).parent / 'models' , 'Model_'), help="Type of model") train_parser.add_argument('--debug', action="store_true", dest="debug", default=False, help="Debug samples.") + train_parser.add_argument('--disable-preview', action="store_true", dest="disable_preview", default=False, help="Disable the preview") train_parser.add_argument('--cpu-only', action="store_true", dest="cpu_only", default=False, help="Train on CPU.") train_parser.add_argument('--force-gpu-idx', type=int, dest="force_gpu_idx", default=-1, help="Force to choose this GPU idx.") diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index d1ff333..9b3fbc8 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -242,4 +242,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() - previewThread (input_queue, output_queue) + if in_options.get('disable_preview', False) == False: + previewThread (input_queue, output_queue)