Errors met when using the correlation package for PyTorch

[shuaiyuan1996]

Hi,

It seems that I have installed the external correlation package for PyTorch successfully (no error messages shown), but when I use the Correlation module under the package, I get the following error.

AttributeError: module 'correlation_package._ext.corr' has no attribute 'corr_cuda_forward'

Does anyone have the same problem? Any ideas on how to fix this?

My software environment: Python 3.6, PyTorch 0.4.0, CUDA 8.0.

Thanks! :)

[ShoufaChen]

I solved this problem by replacing the Correlation package here with the one provided by FlowNet2.

First, download correlation_package to PWC-Net/PyTorch/models/. That looks like:

PWC-Net/PyTorch/models/
    correlation_package
    PWCNet.py
    __init__.py

Then, build the package:

cd  PWC-Net/PyTorch/models/correlation_package
python3 setup.py install --user

Finally, change the import at PWCNet.py line 13

from:

from correlation_package.modules.corr import Correlation 

to:

from .correlation_package.correlation import Correlation

[xiedidan]

I got Segmentation fault (core dumped) with the FlowNet2 Correlation package.
I'm using python 3.7, pytorch 1.4 and cuda 10.1.
Not sure how to solve this...

[teejs2012]

I have encountered error when trying to compile the correlation package. Managed to resolve by compiling it from here (https://github.com/vt-vl-lab/pwc-net.pytorch) and move over the generated folders