- Python 3.6
- PyTorch 1.10.1
- Download the preprocessed data from this link and put in './data' folder.
- the data structure KITRO desires is as following:
{ 'imgname' # image name (list) 'pred_theta' # Predicted 3D rotation matrix (shape: [samples, 24, 3, 3]) 'pred_beta' # Predicted body shape parameters (shape: [samples, 10]) 'pred_cam' # Predicted camera translation (shape: [samples, 3]) 'intrinsics' # Intrinsic camera parameters (shape: [samples, 3, 3]) 'keypoints_2d' # Given 2D keypoints (shape: [samples, 24, 2]) 'GT_pose' # Ground truth 3D rotation parameters (shape: [samples, 72]) 'GT_beta' # Ground truth body shape parameters (shape: [samples, 10]) }
- the data structure KITRO desires is as following:
-
Test on 3DPW
python eval_KITRO.py --data_path 'data/ProcessedData_CLIFFpred_w2DKP_3dpw.pt' >> logs/runkitro_3dpw.out 2>&1
- Corresponding output logs can found at
logs/runkitro_3dpw.out
- Corresponding output logs can found at
-
Test on Human3.6m
python eval_KITRO.py --data_path 'data/ProcessedData_CLIFFpred_w2DKP_HM36.pt' >> logs/runkitro_HM36.out 2>&1
- Corresponding output logs can found at
logs/runkitro_HM36.out
- Corresponding output logs can found at
If you find our paper or codes useful, please consider citing our paper:
Our codes are based on SPIN, CLIFF, and HybrIK and we really appreciate it.