Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Question:result of stage1 model to estimate 3d depth is different? #44

Closed
xinghuokang opened this issue Nov 19, 2018 · 3 comments
Closed

Comments

@xinghuokang
Copy link

Hello, Zhou,
I trained 2D Hourglass Module by your code with 60 epoch, when I was training stage2, the output of depth regress module is very large at the beginning(such as [5000, 122, -13984 ...] and so on) ; But when I used model_60.th you have provided, when I was training stage2, the output of depth regress module is very small at the beginning(such as [0.11, -0.03, 0.43,....., 123]), it doesn't occur large numbers. Why? Look forward your reply.

@xingyizhou
Copy link
Owner

Hi,
Thanks for your interests in this project. This is caused the bug for PyTorch BN bug #16 . Here are two solutions: 1). change pytorch version to 0.1.12. 2). Hack the source code of pytorch BN implementation (https://github.com/Microsoft/human-pose-estimation.pytorch):

# PYTORCH=/path/to/pytorch
# for pytorch v0.4.0
sed -i "1194s/torch\.backends\.cudnn\.enabled/False/g" ${PYTORCH}/torch/nn/functional.py
# for pytorch v0.4.1
sed -i "1254s/torch\.backends\.cudnn\.enabled/False/g" ${PYTORCH}/torch/nn/functional.py

@xinghuokang
Copy link
Author

Thanks for your reply quickly, I will try it. By the way, ECCV2018(Learning 3D Human Pose from Structure and Motion), the paper is researched on your network, What do you think?

@xingyizhou
Copy link
Owner

It's good.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants