Spaces:
Runtime error
Runtime error
File size: 1,790 Bytes
162943d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
"""
This script is borrowed and extended from https://github.com/nkolot/SPIN/blob/master/path_config.py
path configuration
This file contains definitions of useful data stuctures and the paths
for the datasets and data files necessary to run the code.
Things you need to change: *_ROOT that indicate the path to each dataset
"""
import os
# pymaf
pymaf_data_dir = os.path.join(os.path.dirname(__file__),
"../../../data/pymaf_data")
SMPL_MEAN_PARAMS = os.path.join(pymaf_data_dir, 'smpl_mean_params.npz')
SMPL_MODEL_DIR = os.path.join(pymaf_data_dir, '../smpl_related/models/smpl')
CUBE_PARTS_FILE = os.path.join(pymaf_data_dir, 'cube_parts.npy')
JOINT_REGRESSOR_TRAIN_EXTRA = os.path.join(pymaf_data_dir,
'J_regressor_extra.npy')
JOINT_REGRESSOR_H36M = os.path.join(pymaf_data_dir, 'J_regressor_h36m.npy')
VERTEX_TEXTURE_FILE = os.path.join(pymaf_data_dir, 'vertex_texture.npy')
SMPL_MEAN_PARAMS = os.path.join(pymaf_data_dir, 'smpl_mean_params.npz')
SMPL_MODEL_DIR = os.path.join(pymaf_data_dir, '../smpl_related/models/smpl')
CHECKPOINT_FILE = os.path.join(pymaf_data_dir,
'pretrained_model/PyMAF_model_checkpoint.pt')
# pare
pare_data_dir = os.path.join(os.path.dirname(__file__),
"../../../data/pare_data")
CFG = os.path.join(pare_data_dir, 'pare/checkpoints/pare_w_3dpw_config.yaml')
CKPT = os.path.join(pare_data_dir,
'pare/checkpoints/pare_w_3dpw_checkpoint.ckpt')
# hybrik
hybrik_data_dir = os.path.join(os.path.dirname(__file__),
"../../../data/hybrik_data")
HYBRIK_CFG = os.path.join(hybrik_data_dir, 'hybrik_config.yaml')
HYBRIK_CKPT = os.path.join(hybrik_data_dir, 'pretrained_w_cam.pth')
|