diff --git a/.idea/misc.xml b/.idea/misc.xml index 966d4b11..c637d58b 100644 --- a/.idea/misc.xml +++ b/.idea/misc.xml @@ -3,5 +3,5 @@ - + \ No newline at end of file diff --git a/.idea/mmsr.iml b/.idea/mmsr.iml index dc3935a0..2e483868 100644 --- a/.idea/mmsr.iml +++ b/.idea/mmsr.iml @@ -9,7 +9,7 @@ - + diff --git a/codes/process_video.py b/codes/process_video.py index d023495f..650c04cf 100644 --- a/codes/process_video.py +++ b/codes/process_video.py @@ -113,7 +113,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/use_video_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) diff --git a/codes/scripts/byol/byol_extract_wrapped_model.py b/codes/scripts/byol/byol_extract_wrapped_model.py index 90946be6..644ade09 100644 --- a/codes/scripts/byol/byol_extract_wrapped_model.py +++ b/codes/scripts/byol/byol_extract_wrapped_model.py @@ -3,8 +3,8 @@ import torch from models.spinenet_arch import SpineNet if __name__ == '__main__': - pretrained_path = '../../../experiments/resnet_byol_diffframe_115k.pth' - output_path = '../../experiments/resnet_byol_diffframe_115k_.pth' + pretrained_path = '../../../experiments/uresnet_pixpro4_imgset.pth' + output_path = '../../../experiments/uresnet_pixpro4_imgset.pth' wrap_key = 'online_encoder.net.' sd = torch.load(pretrained_path) diff --git a/codes/scripts/extract_square_images.py b/codes/scripts/extract_square_images.py index e02114c2..0859fc41 100644 --- a/codes/scripts/extract_square_images.py +++ b/codes/scripts/extract_square_images.py @@ -13,17 +13,17 @@ import torch def main(): split_img = False opt = {} - opt['n_thread'] = 7 - opt['compression_level'] = 90 # JPEG compression quality rating. + opt['n_thread'] = 3 + opt['compression_level'] = 95 # JPEG compression quality rating. # 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. opt['dest'] = 'file' opt['input_folder'] = ['F:\\4k6k\\datasets\\ns_images\\imagesets\\pn_coven\\working'] - opt['save_folder'] = 'F:\\4k6k\\datasets\\ns_images\\256_unsupervised' - opt['imgsize'] = 256 - opt['bottom_crop'] = 0.1 - opt['keep_folder'] = False + opt['save_folder'] = 'F:\\4k6k\\datasets\\ns_images\\imagesets\\pn_coven\\cropped' + opt['imgsize'] = 1024 + opt['bottom_crop'] = .1 + opt['keep_folder'] = True save_folder = opt['save_folder'] if not osp.exists(save_folder): diff --git a/codes/scripts/extract_temporal_squares.py b/codes/scripts/extract_temporal_squares.py new file mode 100644 index 00000000..e69de29b diff --git a/codes/test.py b/codes/test.py index 6163ef76..b3d4b4bc 100644 --- a/codes/test.py +++ b/codes/test.py @@ -48,7 +48,7 @@ if __name__ == "__main__": torch.backends.cudnn.benchmark = True want_metrics = False parser = argparse.ArgumentParser() - parser.add_argument('-opt', type=str, help='Path to options YAML file.', default='../options/test_faces_styled_sr.yml') + parser.add_argument('-opt', type=str, help='Path to options YAML file.', default='../options/test_mi1.yml') opt = option.parse(parser.parse_args().opt, is_train=False) opt = option.dict_to_nonedict(opt) utils.util.loaded_options = opt diff --git a/codes/train.py b/codes/train.py index fae59c70..fac4fa3a 100644 --- a/codes/train.py +++ b/codes/train.py @@ -295,7 +295,7 @@ class Trainer: if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_imgset_vqvae_stage1.yml') + parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_imgset_tiled_nvqvae_stage1.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()