diff --git a/codes/scripts/extract_subimages.py b/codes/scripts/extract_subimages.py index 0cd225e9..c80142da 100644 --- a/codes/scripts/extract_subimages.py +++ b/codes/scripts/extract_subimages.py @@ -21,8 +21,8 @@ def main(): # compression time. If read raw images during training, use 0 for faster IO speed. if mode == 'single': full_multiplier = .25 - opt['input_folder'] = 'F:\\4k6k\\datasets\\images\\goodeats\\hq\\new_season\\lr_hr_enc\\lr\\images' - opt['save_folder'] = 'F:\\4k6k\\datasets\\images\\goodeats\\hq\\new_season\\lr_hr_enc\\lr\\images_tiled' + opt['input_folder'] = 'F:\\4k6k\\datasets\\images\\fullvideo\\full_images' + opt['save_folder'] = 'F:\\4k6k\\datasets\\images\\fullvideo\\256_tiled' opt['crop_sz'] = int(256 * full_multiplier) # the size of each sub-image opt['step'] = int(128 * full_multiplier) # step of the sliding crop window opt['thres_sz'] = int(64 * full_multiplier) # size threshold diff --git a/codes/scripts/extract_subimages_with_ref.py b/codes/scripts/extract_subimages_with_ref.py index 86f3070d..3466dfbd 100644 --- a/codes/scripts/extract_subimages_with_ref.py +++ b/codes/scripts/extract_subimages_with_ref.py @@ -21,12 +21,12 @@ def main(): if mode == 'single': opt['dest'] = 'file' - 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['input_folder'] = 'F:\\4k6k\\datasets\\images\\fullvideo\\full_images' + opt['save_folder'] = 'F:\\4k6k\\datasets\\images\\fullvideo\\256_tiled' + opt['crop_sz'] = [512, 1024] # the size of each sub-image + opt['step'] = [512, 1024] # step of the sliding crop window opt['thres_sz'] = 128 # size threshold - opt['resize_final_img'] = [1, .5, .25] + opt['resize_final_img'] = [.5, .25] opt['only_resize'] = False save_folder = opt['save_folder'] diff --git a/codes/train.py b/codes/train.py index 4da72be3..122ce55f 100644 --- a/codes/train.py +++ b/codes/train.py @@ -30,7 +30,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_exd_imgset_spsr7.yml') + parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_exd_imgset_chained_structured_trans_invariance.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()