{
'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])
}
Test on 3DPW
python eval_KITRO.py --data_path 'data/ProcessedData_CLIFFpred_w2DKP_3dpw.pt' >> logs/runkitro_3dpw.out 2>&1
logs/runkitro_3dpw.out
Test on Human3.6m
python eval_KITRO.py --data_path 'data/ProcessedData_CLIFFpred_w2DKP_HM36.pt' >> logs/runkitro_HM36.out 2>&1
logs/runkitro_HM36.out
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.