runa91 / barc_release

barc_release
Other
75 stars 9 forks source link

How to train segmentation through an hourglass network? #12

Open G-Apple1 opened 1 year ago

G-Apple1 commented 1 year ago

Thank you for your work!

image

I see you added this paragraph to the hourglass network code, and want to ask if you can provide your complete training process code

This code only trains keypoints:https://github.com/anibali/pytorch-stacked-hourglass

Why do you choose the latter two heatmap? If you train 24 keypoints, do you need to set NUM_JOINT to 26?