How can I get the visible result through your code?

princeton-vl / pytorch_stacked_hourglass

Pytorch implementation of the ECCV 2016 paper "Stacked Hourglass Networks for Human Pose Estimation"

BSD 3-Clause "New" or "Revised" License

469 stars 94 forks source link

How can I get the visible result through your code? #2

Closed WuZheBao0914 closed 4 years ago

WuZheBao0914 commented 4 years ago

Visible result means a pic that painted all the human keypoints.

crockwell commented 4 years ago

There isn't explicit visualization code here, but you can change pixels of the image corresponding to joints to display the keypoints. To do this, for example, you could modify mpii_eval function in test.py to take in images as well as keypoints and write this to tensorboard.