PWC-Net icon indicating copy to clipboard operation
PWC-Net copied to clipboard

Inference time on pytorch version

Open MiaoJieF opened this issue 2 years ago • 6 comments

I use the pytorch version of pwcnet to test the inference time, the image resolution is 1024 x 512, the graphics card used is RTX3060, and the inference time is 0.08s. There is a big gap with the reasoning time in the paper. Is this because the pytorch version is slower than the caffe version?

MiaoJieF avatar Jun 23 '22 12:06 MiaoJieF

I use the pytorch version of pwcnet to test the inference time, the image resolution is 1024 x 512, the graphics card used is RTX3060, and the inference time is 0.08s. There is a big gap with the reasoning time in the paper. Is this because the pytorch version is slower than the caffe version?

hi, i'm having trouble when I configure the python enviroment of the torch version, could you please teach me how to configure the enviroment of the torch version?

andersonhusky avatar Nov 08 '22 13:11 andersonhusky

You can refer to the RAFT environment requirements for configuration.

MiaoJieF avatar Nov 08 '22 13:11 MiaoJieF

You can refer to the RAFT environment requirements for configuration

I'm sorry, what is "the RAFT environment "? I follw the steps in Torch/readme.md, but it seems like cuda 11.2 of 3060 doesnt match with the cuda the code need..

andersonhusky avatar Nov 08 '22 14:11 andersonhusky

1.RAFT is a well-known optical flow estimation algorithm which is also open source. 2.The cuda version I use is 11.1. The Pytorch version is 1.9.1. 3.Also, see your question under Flowformer. That's due to a lack of memory on the GPU, try using the lightweight version provided by the author.

MiaoJieF avatar Nov 08 '22 14:11 MiaoJieF

1.RAFT is a well-known optical flow estimation algorithm which is also open source. 2.The cuda version I use is 11.1. The Pytorch version is 1.9.1. 3.Also, see your question under Flowformer. That's due to a lack of memory on the video card, try using the lightweight version provided by the author.

Thanks! You saved my day!

andersonhusky avatar Nov 08 '22 14:11 andersonhusky

1.RAFT is a well-known optical flow estimation algorithm which is also open source. 2.The cuda version I use is 11.1. The Pytorch version is 1.9.1. 3.Also, see your question under Flowformer. That's due to a lack of memory on the GPU, try using the lightweight version provided by the author.

Hi, sorry to bother you again. I have successfully configure the enviroment of RAFT, the problem is RAFT is based on python3, and I still cant run pwcnet-torch in this enviroment successfully... by the way, here is my email: [email protected], if you feel inconvenient to explain here, we can communicate via email, thanks for you generous help again!

andersonhusky avatar Nov 09 '22 05:11 andersonhusky