search

sniklaus / pytorch-pwc

a reimplementation of PWC-Net in PyTorch that matches the official Caffe version
GNU General Public License v3.0
608 stars 122 forks source link

KITTI finetune model #29

Open htkseason opened 4 years ago

htkseason commented 4 years ago

Hi Niklaus, Thanks for providing a clean implementaion of pwc-net on pytorch. I am working with your code on KITTI dataset. However, its performance did not match PWC-Net and PWC-Net+. I assume the model is not finetuned on KITTI and not using the training strategy of PWC-Net+. Refer https://github.com/NVlabs/PWC-Net/tree/master/Caffe/model/PWC-Net_plus Could you please also convert the model of the above link or share your conversion script?

sniklaus commented 4 years ago

Please see: #23

I am afraid that I have not made any efforts to adopt PWC-Net+ yet. I am thus leaving this issue open since I agree that it would be nice to have those models as well.

zhm284 commented 3 months ago

What should I do if my picture is too large by 1920 * 3840

Etienne66 commented 3 months ago

You can either scale down your image so that it fits in memory or get a video card that has more memory. I use an RTX 3090 myself.