search

dineshreddy91 / Occlusion_Net

[CVPR2019]Occlusion-Net: 2D/3D Occluded Keypoint Localization Using Graph Networks
Other
139 stars 34 forks source link

Person keypoint detection not working #13

Closed AkhileshV closed 4 years ago

AkhileshV commented 4 years ago

Occlusion_Net/lib/predictor.py", line 617, in vis_keypoints_person p2 = kps[0, i2], kps[1, i2] IndexError: index 14 is out of bounds for axis 1 with size 14

dineshreddy91 commented 4 years ago

The trained network currently only supports vehicle keypoints. Person keypoints is not yet incorporated into the framework.

leon532 commented 4 years ago

@dineshreddy91 Thanks for sharing your great work. Did you test the performance of OcclusionNet on person keypoint datasets? How is the effect?

dineshreddy91 commented 4 years ago

The theory in general should work great for humans as well. The datasets are widely available for persons online. So all you have to do is train the method on the dataset.

leon532 commented 4 years ago

@dineshreddy91 Thanks.