princeton-vl / pose-hg-demo

Code to test and use the model from "Stacked Hourglass Networks for Human Pose Estimation"
BSD 3-Clause "New" or "Revised" License
316 stars 86 forks source link

A quarter of offest? #16

Open hellojialee opened 6 years ago

hellojialee commented 6 years ago

In the original paper, there goes: "the prediction is offset by a quarter of a pixel in the direction of its next highest neighbor before transforming back to the original coordinate space of the image"

Is it implemented in this project? Looking forward for replying. Thank you!

anewell commented 6 years ago

I don't think it is included here, but you can find the code for it in the training repo here: https://github.com/umich-vl/pose-hg-train/blob/master/src/util/pose.lua#L85

hellojialee commented 6 years ago

@anewell Thank you for your kind replying! I'm not familiar with lua. Do you simply find the max response (regarded as a pixel value, let's denote it P) on the heat map and then shift towards to the second max response which is within its 4-neighbors or just another max response of a gussian circle? image