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

Heatmap Visualization #67

Closed A7777-gp closed 1 year ago

A7777-gp commented 1 year ago

Hello, what part of the code is the visualization of the heat map? How can I get it?

hellojialee commented 1 year ago

In this repo, I use the stupid method: https://github.com/hellojialee/Improved-Body-Parts/blob/0fa17dff1ea829c2951c18185d5f1ce7fe0072fc/demo_image.py#L64

A7777-gp commented 1 year ago

Thanks!!! But in demo_image.py, the last output is the human body posture, may I ask which line of code in demo_image.py can be modified to output the heat map of key points?

hellojialee commented 1 year ago

Here: https://github.com/hellojialee/Improved-Body-Parts/blob/0fa17dff1ea829c2951c18185d5f1ce7fe0072fc/demo_image.py#L98

You can specify which keypoint heatmap to show by its index.

A7777-gp commented 1 year ago

May I ask if the key point heat map can be output by executing this line of code? https://github.com/hellojialee/Improved-Body-Parts/blob/0fa17dff1ea829c2951c18185d5f1ce7fe0072fc/demo_image.py#L191

A7777-gp commented 1 year ago

I'm sorry. I'm a terrible coder😭

hellojialee commented 1 year ago

May I ask if the key point heat map can be output by executing this line of code?

https://github.com/hellojialee/Improved-Body-Parts/blob/0fa17dff1ea829c2951c18185d5f1ce7fe0072fc/demo_image.py#L191

Yes. No apologies ：）