|
""" |
|
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_data_dir = os.path.join(os.path.dirname(__file__), |
|
"../../../data/HPS/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") |
|
MESH_DOWNSAMPLEING = os.path.join(pymaf_data_dir, "mesh_downsampling.npz") |
|
|
|
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") |
|
CHECKPOINT_FILE = os.path.join(pymaf_data_dir, |
|
"pretrained_model/PyMAF_model_checkpoint.pt") |
|
|
|
|
|
pare_data_dir = os.path.join(os.path.dirname(__file__), |
|
"../../../data/HPS/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_data_dir = os.path.join(os.path.dirname(__file__), |
|
"../../../data/HPS/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") |
|
|
