search

hellojialee / Improved-Body-Parts

Simple Pose: Rethinking and Improving a Bottom-up Approach for Multi-Person Pose Estimation
https://arxiv.org/abs/1911.10529
258 stars 42 forks source link

Understanding body part heatmap implementation #44

Closed jysa01 closed 3 years ago

jysa01 commented 3 years ago

@hellojialee , Kudos to the excellent work,

I am referring to your ground truth heatmap generation implementation because I must generate body part heatmaps for my study. In short, I am looking to achieve this : image

I intend to generate 18 channel heatmap where each channel stores one body part ( line segment between wrist and elbow, line segment between elbow and shoulder etc ) It is not clear to me how the elliptical gaussian has been implemented. Can you please explain the steps in put_limb_gaussian_maps in /py_cocodata_server/py_data_heatmapper.py because I believe this is the function doing my desired task.

Thanks in advance, jysa01

hellojialee commented 3 years ago

Hi jysa01, thank you for your interest!

  1. Calculate the vertical distance (for example d) from a point P to the major axis of a body part
  2. Compute the Gaussian response using exp(- d / 2sigma^2)
  3. Set the heatmap pixels outside the elliptical area to zero
jysa01 commented 3 years ago

Thank you for the explanation @hellojialee .