mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Misc

Browse Source
This commit is contained in:
James Betker 2020-10-20 12:56:52 -06:00
parent aba83e7497
commit d8c6a4bbb8
3 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/scripts/extract_subimages.py
View File

@ -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

10
codes/scripts/extract_subimages_with_ref.py
View File

@ -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']

2
codes/train.py
View File

@ -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()