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

save configs

Browse Source
This commit is contained in:
luni
2022-10-16 23:27:24 +00:00
parent 3a79508a8c
commit 8eaf090e4c
6 changed files with 77 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/config_pyramid.py
+12 -9
View File
@@ -1,7 +1,7 @@
from configs.config_base import config
config['gpu'] = '1'
config['name'] = 'cvae_pyr_7_32_64_128_256_512'
config['gpu'] = '3'
config['name'] = 'pyr_1_32_64_128'
config['notes'] = '\
offsets over average unposed garment. remove global orientation (in posed body input and in pose_shape parameters. | \
retrained body encoder with deoriented bodies | \
@@ -56,9 +56,9 @@ config['pyramid'] = dict()
config['pyramid']['is_label'] = False
config['pyramid']['latent_dim'] = config['latent_dim']
config['pyramid']['resolutions'] = [32, 64, 128, 256, 512]
config['pyramid']['resolutions'] = [32, 64, 128]
config['pyramid']['loss_weights'] = [1,1,1,1,1]
config['pyramid']['pretrained_pyramid'] = 'checkpoints/cvae_pyr_7_32_64_128_256_50k'
config['pyramid']['pretrained_pyramid'] = 'checkpoints/pyr_1_32_64'
for res in config['pyramid']['resolutions']:
if res != 512:
@@ -73,13 +73,16 @@ for res in config['pyramid']['resolutions']:
# config['pyramid']['static_model'] = 'checkpoints/ae_static_7_2_x'
# config['pyramid']['body_model'] = 'checkpoints/ae_body_9_x'
# config['pyramid']['curvature_model'] = 'checkpoints/ae_curv_9_x'
# config['pyramid']['static_model'] = 'checkpoints/static_1'
# config['pyramid']['body_model'] = 'checkpoints/body_1'
# config['pyramid']['curvature_model'] = 'checkpoints/curv_1'
config['pyramid']['static_model'] = ''
config['pyramid']['body_model'] = ''
config['pyramid']['curvature_model'] = ''
config['pyramid']['freeze_encoders'] = True
config['avg_garment_path'] = ''
config['pyramid']['final_upscale_enhancer'] = ''
config['pyramid']['final_upscale_enhancer'] = 'checkpoints/128_512_3'
config['pyramid']['levels'] = []
@@ -108,7 +111,7 @@ for res in config['pyramid']['resolutions']:
level['decoder_depths'] = [9, 9, 9, 9]
level['decoder_dims'] = [256, 128, 64, 16]
level['avg_garment_path'] = ''
level['upscale_enhancer'] = 'checkpoints/up_3_d256_512_2'
level['upscale_enhancer'] = ''#'checkpoints/up_3_d256_512_2'
level['upscale_enhancer_freeze'] = True
config['pyramid']['levels'].append(level)
elif res == 128:
@@ -122,7 +125,7 @@ for res in config['pyramid']['resolutions']:
level['decoder_depths'] = [16, 16, 9, 9]
level['decoder_dims'] = [256, 256, 128, 64]
level['avg_garment_path'] = ''
level['upscale_enhancer'] = 'checkpoints/up_3_d128_256_2'
level['upscale_enhancer'] = ''#'checkpoints/up_3_d128_256_2'
level['upscale_enhancer_freeze'] = True
config['pyramid']['levels'].append(level)
elif res == 64:
@@ -136,7 +139,7 @@ for res in config['pyramid']['resolutions']:
level['decoder_depths'] = [27, 27, 16]
level['decoder_dims'] = [256, 256, 256]
level['avg_garment_path'] = ''
level['upscale_enhancer'] = 'checkpoints/up_3_d64_128_2'
level['upscale_enhancer'] = 'checkpoints/64_512_2'#checkpoints/up_3_d64_128_2'
level['upscale_enhancer_freeze'] = False
config['pyramid']['levels'].append(level)
elif res == 32:
@@ -150,7 +153,7 @@ for res in config['pyramid']['resolutions']:
level['decoder_depths'] = [27, 27]
level['decoder_dims'] = [256, 256]
level['avg_garment_path'] = ''
level['upscale_enhancer'] = 'checkpoints/up_3_d32_64_2'
level['upscale_enhancer'] = 'checkpoints/32_512_4'#'checkpoints/up_3_d32_64_2'
level['upscale_enhancer_freeze'] = True
config['pyramid']['levels'].append(level)
else: