39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
from utils.convnext_utils import CNBlockConfig
|
|
|
|
config = dict()
|
|
|
|
config['batch_size'] = 4
|
|
config['is_cape'] = False
|
|
config['device'] = 'cuda'
|
|
|
|
config['train_dir'] = '/data-local/127/cloth3d_processed'
|
|
config['valid_dir'] = '/data-local/127/cloth3d_processed'
|
|
config['test_dir'] = '/data-local/127_2/cloth3d_processed_test'
|
|
|
|
config['signature']: set = { 'id',
|
|
'frame_nr',
|
|
'pose_shape',
|
|
'uv_body_posed',
|
|
'mesh_body_posed',
|
|
'mesh_posed',
|
|
'mesh_unposed',
|
|
'uv_unposed',
|
|
'uv_posed',
|
|
'uv_normals',
|
|
'uv_static',
|
|
'mesh_mask',
|
|
'uv_mask',
|
|
}
|
|
|
|
|
|
|
|
for res in [64, 128, 256]:
|
|
res_modifier = '_' + str(res)
|
|
config['signature'].add('uv_body_posed' + res_modifier)
|
|
config['signature'].add('uv_unposed' + res_modifier)
|
|
config['signature'].add('uv_posed' + res_modifier)
|
|
config['signature'].add('uv_normals' + res_modifier)
|
|
config['signature'].add('uv_static' + res_modifier)
|
|
config['signature'].add('uv_mask' + res_modifier)
|
|
|