-
Notifications
You must be signed in to change notification settings - Fork 38
/
config.py
46 lines (33 loc) · 2.2 KB
/
config.py
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
40
41
42
43
44
45
46
r"""
Config for paths, joint set, and normalizing scales.
"""
# datasets (directory names) in AMASS
# e.g., for ACCAD, the path should be `paths.raw_amass_dir/ACCAD/ACCAD/s001/*.npz`
amass_data = ['HumanEva', 'MPI_HDM05', 'SFU', 'MPI_mosh', 'Transitions_mocap', 'SSM_synced', 'CMU',
'TotalCapture', 'Eyes_Japan_Dataset', 'KIT', 'BMLmovi', 'EKUT', 'TCD_handMocap', 'ACCAD',
'BioMotionLab_NTroje', 'BMLhandball', 'MPI_Limits', 'DFaust67']
class paths:
raw_amass_dir = 'data/dataset_raw/AMASS' # raw AMASS dataset path (raw_amass_dir/ACCAD/ACCAD/s001/*.npz)
amass_dir = 'data/dataset_work/AMASS' # output path for the synthetic AMASS dataset
raw_dipimu_dir = 'data/dataset_raw/DIP_IMU' # raw DIP-IMU dataset path (raw_dipimu_dir/s_01/*.pkl)
dipimu_dir = 'data/dataset_work/DIP_IMU' # output path for the preprocessed DIP-IMU dataset
# DIP recalculates the SMPL poses for TotalCapture dataset. You should acquire the pose data from the DIP authors.
raw_totalcapture_dip_dir = 'data/dataset_raw/TotalCapture/DIP_recalculate' # contain ground-truth SMPL pose (*.pkl)
raw_totalcapture_official_dir = 'data/dataset_raw/TotalCapture/official' # contain official gt (S1/acting1/gt_skel_gbl_pos.txt)
totalcapture_dir = 'data/dataset_work/TotalCapture' # output path for the preprocessed TotalCapture dataset
result_dir = 'data/result' # output directory for the evaluation results
smpl_file = 'models/SMPL_male.pkl' # official SMPL model path
physics_model_file = 'models/physics.urdf' # physics body model path
plane_file = 'models/plane.urdf' # (for debug) path to plane.urdf Please put plane.obj next to it.
weights_file = 'data/weights.pt' # network weight file
physics_parameter_file = 'physics_parameters.json' # physics hyperparameters
class joint_set:
leaf = [7, 8, 12, 20, 21]
full = list(range(1, 24))
reduced = [1, 2, 3, 4, 5, 6, 9, 12, 13, 14, 15, 16, 17, 18, 19]
ignored = [0, 7, 8, 10, 11, 20, 21, 22, 23]
n_leaf = len(leaf)
n_full = len(full)
n_reduced = len(reduced)
n_ignored = len(ignored)
vel_scale = 3