luni/cvpr_2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

config updates

Browse Source
This commit is contained in:
luni
2023-07-06 16:34:15 +02:00
parent 37ad3a3f43
commit 639af22b48
25 changed files with 3620 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/iccvw/pyr_64_cape_f.py
+7 -5
View File
@@ -33,6 +33,8 @@ config['valid_dir'] = '/data-local/127_2/cape_processed/test/female'
config['test_dir'] = '/data-local/127_2/cape_processed/test/female'
config['train_frames'] = None
config['valid_frames'] = None
config['test_frames'] = None
config['test_frames'] = None
config['dataset_in_memory'] = False
config['resolution'] = None
@@ -117,7 +119,7 @@ for res in config['pyramid']['resolutions']:
level = dict()
level['resolution'] = 256
level['kernel_size'] = 5
level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_256'
level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_256_cape_f'
level['upscale_enhancer_freeze'] = True
level['bottleneck_size'] = 64 * config['bottleneck_scale']
level['input_features'] = 3+3+3+3 if config['use_normals'] else 3+3+3
@@ -132,7 +134,7 @@ for res in config['pyramid']['resolutions']:
block_setting_reverse = [
CNBlockConfig(level['bottleneck_size'] * 2, 128, 3),
CNBlockConfig(128, 128, 9),
CNBlockConfig(128, 65, 3),
CNBlockConfig(128, 64, 3),
CNBlockConfig(64, 3, 3),
]
@@ -143,7 +145,7 @@ for res in config['pyramid']['resolutions']:
level = dict()
level['resolution'] = 128
level['kernel_size'] = 5
level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_128'
level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_128_cape_f'
level['upscale_enhancer_freeze'] = True
level['bottleneck_size'] = 64 * config['bottleneck_scale']
level['input_features'] = 3+3+3+3 if config['use_normals'] else 3+3+3
@@ -156,7 +158,7 @@ for res in config['pyramid']['resolutions']:
block_setting_reverse = [
CNBlockConfig(level['bottleneck_size'] * 2, 128, 9),
CNBlockConfig(128, 128, 9),
CNBlockConfig(128, 64, 9),
CNBlockConfig(64, 3, 9),
]
@@ -167,7 +169,7 @@ for res in config['pyramid']['resolutions']:
level = dict()
level['resolution'] = 64
level['kernel_size'] = 3
level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_64'
level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_64_cape_f'
level['upscale_enhancer_freeze'] = True
level['bottleneck_size'] = 64 * config['bottleneck_scale']
level['input_features'] = 3+3+3+3 if config['use_normals'] else 3+3+3