proplem of Integral Pose Regression - Githubissues

JimmySuen / integral-human-pose

Integral Human Pose Regression

MIT License

471 stars 76 forks source link

proplem of Integral Pose Regression #42

Open wangm-word opened 3 years ago

wangm-word commented 3 years ago

thank you for your great work! I want to know,when you train it ?the 2D data miss "Z" coordinate value,how to deal with it?

lck1201 commented 3 years ago

generate a mask applying on Z, see https://github.com/JimmySuen/integral-human-pose/blob/ad3f875bc05538da3471ef81484e23fad3e9c787/pytorch_projects/common_pytorch/dataset/mpii.py#L171

https://github.com/JimmySuen/integral-human-pose/blob/ad3f875bc05538da3471ef81484e23fad3e9c787/pytorch_projects/common_pytorch/common_loss/weighted_mse.py#L13

in this way, loss of z won't affect BP.