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-06-02 09:35:52 -06:00
parent 726d1913ac
commit 76a38b6a53
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
codes/data_scripts/extract_subimages.py
View File

@ -20,10 +20,10 @@ def main():
# CV_IMWRITE_PNG_COMPRESSION from 0 to 9. A higher value means a smaller size and longer
# compression time. If read raw images during training, use 0 for faster IO speed.
if mode == 'single':
opt['input_folder'] = 'F:\\4k6k\\datasets\\sft_ns_images\\images'
opt['save_folder'] = 'F:\\4k6k\\datasets\\sft_ns_images\\images_tiled'
opt['crop_sz'] = 300 # the size of each sub-image
opt['step'] = 280 # step of the sliding crop window
opt['input_folder'] = 'F:\\4k6k\\datasets\\imagesets\\new'
opt['save_folder'] = 'F:\\4k6k\\datasets\\imagesets\\unfiltered_tiled_2x'
opt['crop_sz'] = 512 # the size of each sub-image
opt['step'] = 440 # step of the sliding crop window
opt['thres_sz'] = 120 # size threshold
extract_single(opt, split_img)
elif mode == 'pair':

2
codes/distill_torchscript.py
View File

@ -9,7 +9,7 @@ import torch.nn.functional as F
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('-opt', type=str, help='Path to options YAML file.', default='../options/use_vrp_upsample.yml')
parser.add_argument('-opt', type=str, help='Path to options YAML file.', default='../options/use_video_upsample.yml')
opt = option.parse(parser.parse_args().opt, is_train=False)
opt = option.dict_to_nonedict(opt)
netG = define_G(opt)

2
codes/test.py
View File

@ -17,7 +17,7 @@ if __name__ == "__main__":
torch.backends.cudnn.benchmark = True
want_just_images = True
parser = argparse.ArgumentParser()
parser.add_argument('-opt', type=str, help='Path to options YMAL file.', default='../options/upsample_full_movie.yml')
parser.add_argument('-opt', type=str, help='Path to options YMAL file.', default='../options/test_resgen_upsample.yml')
opt = option.parse(parser.parse_args().opt, is_train=False)
opt = option.dict_to_nonedict(opt)