ilovepose / DarkPose

Distribution-Aware Coordinate Representation for Human Pose Estimation
https://ilovepose.github.io/coco
Apache License 2.0
550 stars 80 forks source link

the code has a little flaw #24

Closed seekFire closed 3 years ago

seekFire commented 3 years ago

In inference.py, the function get_max_preds, I think the code shown as follows needs to add - 1 to its tail: preds[:, :, 0] = (preds[:, :, 0]) % width should be rectified as: preds[:, :, 0] = (preds[:, :, 0]) % width - 1 So what's your opinion?

seekFire commented 3 years ago

I made a mistake... you're right