search

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
465 stars 94 forks source link

coordinate overlap #47

Closed Winnie202 closed 8 months ago

Winnie202 commented 10 months ago

Thank you for your contribution of the code. I completed the detection of 4 key points according to your code, but the second coordinate and the third coordinate overlap. May I ask why， How to solve this problem

2023-09-22 18-35-10屏幕截图

crockwell commented 10 months ago

Hi Winnie,

This model was trained on predicting joints in humans. Therefore, I wouldn't expect it to be best for generic detection of keypoints e.g. of a piece of paper. I'd recommend using a generic detection method such as the YOLO line of work.

Regards, Chris