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

Do you have a model trained on Kitti dataset? #9

Closed
ZheC opened this issue Oct 12, 2018 · 6 comments
Closed

Do you have a model trained on Kitti dataset? #9

ZheC opened this issue Oct 12, 2018 · 6 comments

Comments

@ZheC
Copy link

ZheC commented Oct 12, 2018

Thank you for the re-implementation! It is so helpful. Do you have a model trained on the Kitti flow 2015 dataset?

@sniklaus
Copy link
Owner

I am afraid that an official model trained on KITTI has not been released yet [1]. Since I am trying to stick as close as possible to the original repository, and also did not train on KITTI myself, I have not planned to release such a model. My apologies, but thank you for being interested in this implementation.

[1] https://github.com/NVlabs/PWC-Net/tree/master/Caffe/model

@ZheC
Copy link
Author

ZheC commented Oct 12, 2018

I think the pwc_net.pth.tar in the original repo is finetuned on kitti dataset as well [1]. Which model in your repo corresponds to pwc_net.pth.tar?

[1] NVlabs/PWC-Net#26

@sniklaus
Copy link
Owner

sniklaus commented Oct 12, 2018

--model sintel corresponds to pwc_net.caffemodel
--model chairs-things corresponds do pwc_net_chairs_things.caffemodel

I should probably change --model sintel to --model various since Deqing mentioned that it has been fine-tuned on a diverse set of training data: NVlabs/PWC-Net#26 (comment)

Thank you for bringing this to my attention.

@sniklaus
Copy link
Owner

I just changed --model sintel to --model default. Thank you again for bringing this up.

@SteveSZF
Copy link

Is the KITTI model only fine-tuned on KITTI2015? not a mixture of KITTI2012 and 2015?

@sniklaus
Copy link
Owner

According to NVlabs/PWC-Net#26 (comment), yes. You might want to consider asking your question there to get a definitive answer.

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

3 participants