From eb12b5f887897669d7949ac5a571685d6cc648f3 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 26 Sep 2020 21:27:17 -0600 Subject: [PATCH] Misc --- codes/data_scripts/extract_subimages_with_ref.py | 4 ++-- codes/train.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/codes/data_scripts/extract_subimages_with_ref.py b/codes/data_scripts/extract_subimages_with_ref.py index 05864c98..9c69318f 100644 --- a/codes/data_scripts/extract_subimages_with_ref.py +++ b/codes/data_scripts/extract_subimages_with_ref.py @@ -23,8 +23,8 @@ def main(): if mode == 'single': opt['dest'] = 'file' - opt['input_folder'] = 'F:\\4k6k\\datasets\\ns_images\\imagesets\\images-half' - opt['save_folder'] = 'F:\\4k6k\\datasets\\ns_images\\imagesets\\new_tiles' + opt['input_folder'] = 'F:\\4k6k\\datasets\\ns_images\\vixen\\full_video_segments' + opt['save_folder'] = 'F:\\4k6k\\datasets\\ns_images\\vixen\\full_video_256_tiled_with_ref' opt['crop_sz'] = [256, 512, 1024] # the size of each sub-image opt['step'] = [256, 512, 1024] # step of the sliding crop window opt['thres_sz'] = 128 # size threshold diff --git a/codes/train.py b/codes/train.py index 53c0c4f2..f88e327c 100644 --- a/codes/train.py +++ b/codes/train.py @@ -32,7 +32,7 @@ def init_dist(backend='nccl', **kwargs): def main(): #### options parser = argparse.ArgumentParser() - parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_blur_discriminator.yml') + parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_exd_imgset_spsr5_spine_no_pretrain.yml') parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher') parser.add_argument('--local_rank', type=int, default=0) args = parser.parse_args()