search

JiaRenChang / PSMNet

Pyramid Stereo Matching Network (CVPR2018)
MIT License
1.44k stars 423 forks source link

Would you please provides the PSMNet model pertrained on Scene Flow #2

Closed j-c-zhang closed 6 years ago

j-c-zhang commented 6 years ago

Hello, Jia-Ren Chang.

I want to conduct some experiments based on your work. However, I have trouble in reproducing your experiment result on KITTI Stereo Evaluation 2015 due to the lack of PSMNet model pertrained on Scene Flow dataset. Would you please provides the PSMNet model pertrained on Scene Flow for me?

Thank you very much!

JiaRenChang commented 6 years ago

Hi, We shared the model that we used on KITTI 2015. Please refer to README. Jia-Ren

j-c-zhang commented 6 years ago

I got it! Thank you!

scholltan commented 6 years ago

@j-c-zhang have you decompressed the .tar file correctly? I got the following error: tar -xvf pretrained_model_KITTI2015.tar tar: This does not look like a tar archive tar: Skipping to next header tar: Exiting with failure status due to previous errors

JiaRenChang commented 6 years ago

@scholltan You can use torch.load() to load the model

bryanyzhu commented 6 years ago

@JiaRenChang Hi, thanks for the wonderful work. I understand that you already provide the trained model for KITTI2015. But is it ok to share your pre-trained model on SceneFlow dataset as well? Because not all people are working on KITTI, a pre-trained model on SceneFlow would be much more helpful. Thank you.