search

Jeff-sjtu / HybrIK

Official code of "HybrIK: A Hybrid Analytical-Neural Inverse Kinematics Solution for 3D Human Pose and Shape Estimation", CVPR 2021
MIT License
1.21k stars 147 forks source link

AssertionError: 'img_center' #116

Open CHENYUHANG-smp opened 1 year ago

CHENYUHANG-smp commented 1 year ago

Hello, When I train the model with predicted camera as your step, there is a AssertionError: hybrik_problem It seems that the data domain miss a key 'img_center' which is included in the target. I add the key to data domain but it seems not work. How can I solve this problem. hybrik_pro3

CHENYUHANG-smp commented 1 year ago

When I simply added the key 'img_center' to the data domain, it threw the error while enumerating: image

I thought this error is caused by the 2d dataset MSCOCO.

I first checked the data transform code when training without predicted camera, I found that you bring the keys of 2D and 3D datasets into correspondence, and patch up the keys which 2D datasets missed. image

But when you training the model with predicted camera, it seems that the keys is not matching. image

And the key 'target_xyz' and 'target_xyz_weight' are also not included in the 3D SMPL CAM datasets, so its not necessary. But the key 'img_center' is included in 3D datasets, it's necessary. I thought that's why it threw the error above.

anymyb commented 1 year ago

@CHENYUHANG-smp, @Jeff-sjtu I met the same problem. Have you solved it?