diff --git a/configs/iccv/config_abl_pyr_32_skin.py b/configs/iccv/config_abl_pyr_32_skin.py index 27e8512..b339dc9 100644 --- a/configs/iccv/config_abl_pyr_32_skin.py +++ b/configs/iccv/config_abl_pyr_32_skin.py @@ -58,6 +58,11 @@ config['signature_dynamic'] = { 'id': (), 'frame_nr': (), 'uv_unposed': (512, 256, 3), 'uv_curvature': (512, 256, 7), } +if config['use_skinning']: + config['signature_dynamic']['uv_unposed'] = (512, 256, 3) + config['signature_dynamic']['mesh_unposed'] = (config['smpl_res'], 3,) +else: + config['signature_dynamic']['uv_posed'] = (512, 256, 3) config['signature_static'] = { 'uv_static': (512, 256, 3), 'mesh_mask': (config['smpl_res'],), 'uv_mask': (512, 256,), @@ -73,12 +78,14 @@ for res in config['pyramid']['resolutions']: if res != 512: res_modifier = '_' + str(res) config['signature_dynamic']['uv_body_posed' + res_modifier] = (res, res//2, 3) - config['signature_dynamic']['uv_unposed' + res_modifier] = (res, res//2, 3) + if config['use_skinning']: + config['signature_dynamic']['uv_unposed' + res_modifier] = (res, res//2, 3) + else: + config['signature_dynamic']['uv_posed' + res_modifier] = (res, res//2, 3) config['signature_dynamic']['uv_curvature' + res_modifier] = (res, res//2, 7) config['signature_static']['uv_static' + res_modifier] = (res, res//2, 3) config['signature_static']['uv_mask' + res_modifier] = (res, res//2) - config['pyramid']['static_model'] = 'checkpoints/iccv/static_iccv_1' config['pyramid']['body_model'] = 'checkpoints/iccv/body_iccv_1' config['pyramid']['curvature_model'] = 'checkpoints/iccv/curv_iccv_1'