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