pytorch-pwc icon indicating copy to clipboard operation
pytorch-pwc copied to clipboard

KITTI finetune model

Open htkseason opened this issue 6 years ago • 4 comments

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?

htkseason avatar Sep 25 '19 19:09 htkseason

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.

sniklaus avatar Dec 04 '19 20:12 sniklaus

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

zhm284 avatar Apr 02 '24 07:04 zhm284

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.

Etienne66 avatar Apr 06 '24 02:04 Etienne66