search

facebookresearch / DensePose

A real-time approach for mapping all human pixels of 2D RGB images to a 3D surface-based model of the body
http://densepose.org
Other
6.96k stars 1.29k forks source link

Getting Keypoints #76

Open fire17 opened 6 years ago

fire17 commented 6 years ago

Hi there! Been playing around with this for a while now... All I'm able to do is to get the infer and simple_infer to give me heat maps (which is still amazing! but) all I'm interested in are the Keypoints. Using any other configs and models give me errors. (also tried running detectron but that didn't work either ☹️)

The goal is to get an output file (json for example) of humans and their keypoints or rather find where this data is accessible in the code... Has anyone been able to actually do this?? A bunch of others looked for this in other issues, but havn't found any clear answers though...

A detailed guild / a working model+config will be more than appreciated 😄 Please help! Thank you very much Tami

vkhalidov commented 6 years ago

@fire17 To get keypoints, you need to train the model with the keypoints head. Please have a look at #34, #39, #48