PWC-Net(PyTorch v1.0.1)
Pytorch实现 。 我们将其作为现成的软件包提供:
安装后,只需将整个文件夹PWC_src复制到您的代码库即可使用。 有关详细信息,请参见demo.py。
环境
此代码已经过Python3.6和PyTorch1.0.1和Tesla K80 GPU的测试。 系统为Ubuntu 14.04,CUDA版本为10.0。 所有必需的python包都可以在requirements.txt找到。
安装
# install custom layers
cd PWC_src/correlation_package
python setup.py install
注意:根据您使用的GPU,您可能需要在添加gencode 。 您可以和找到有关gencode更多信息。
转换后的Caffe预训练模型
您可以在models文件夹中找到它们。
推论模式
修改输入的
1