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-09-26 21:27:17 -06:00
parent 31641d7f63
commit eb12b5f887
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/data_scripts/extract_subimages_with_ref.py
View File

@ -23,8 +23,8 @@ def main():
if mode == 'single': if mode == 'single':
opt['dest'] = 'file' opt['dest'] = 'file'
opt['input_folder'] = 'F:\\4k6k\\datasets\\ns_images\\imagesets\\images-half' opt['input_folder'] = 'F:\\4k6k\\datasets\\ns_images\\vixen\\full_video_segments'
opt['save_folder'] = 'F:\\4k6k\\datasets\\ns_images\\imagesets\\new_tiles' 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['crop_sz'] = [256, 512, 1024] # the size of each sub-image
opt['step'] = [256, 512, 1024] # step of the sliding crop window opt['step'] = [256, 512, 1024] # step of the sliding crop window
opt['thres_sz'] = 128 # size threshold opt['thres_sz'] = 128 # size threshold

2
codes/train.py
View File

@ -32,7 +32,7 @@ def init_dist(backend='nccl', **kwargs):
def main(): def main():
#### options #### options
parser = argparse.ArgumentParser() 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('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher')
parser.add_argument('--local_rank', type=int, default=0) parser.add_argument('--local_rank', type=int, default=0)
args = parser.parse_args() args = parser.parse_args()