search

NVlabs / PWC-Net

PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume, CVPR 2018 (Oral)
Other
1.64k stars 355 forks source link

ImportError: No module named _corr #71

Open ZeeshanNadir opened 5 years ago

ZeeshanNadir commented 5 years ago

After successfully building the external correlation package for pytorch, when I run the PWC-Net I get the following error: File "./correlation_package/ext/corr/__init_\.py", line 3, in from ._corr import lib as _lib, ffi as _ffi ImportError: No module named _corr

Looking inside the folders, I don't see any module "_corr". Can you please help me resolve this.

Python version 2.7 PyTorch 0.2

wyhcqq commented 5 years ago

I also meet this problem， have you solve that？

wyhcqq commented 5 years ago

if you solved this, tell me, please

HeliosZhao commented 5 years ago

I meet the same problem, and have no idea how to solve it

jeffbaena commented 5 years ago

Hi all, I also run into the same problem and at the end I went for this version of pwc (PWC-net_ROB) which uses cupy and weighted ported from caffe https://github.com/sniklaus/pytorch-pwc

For the correlation layer another hint might be using the implementation of https://github.com/ClementPinard/FlowNetPytorch , I believe it is well made and in the repository there are also some links to the original nvidia impelementation. You have to skim a bit through the rep to find everything, but I have used it and it is good.

I hope this helps, Stefano