【文件属性】：
文件名称：pwc-net.pytorch:PyTorch-1.0 ＋中的现成PWC-Net模块
文件大小：65.55MB
文件格式：ZIP
更新时间：2021-05-04 19:17:21
optical-flow Python
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文件夹中找到它们。
推论模式
修改输入的
【文件预览】：
pwc-net.pytorch-master
----.gitignore(33B)
----requirements.txt(213B)
----models()
--------sintel.pytorch(35.79MB)
--------chairs-things.pytorch(35.79MB)
----LICENSE(7KB)
----PWC_src()
--------pwc.py(11KB)
--------flowlib.py(14KB)
--------__init__.py(48B)
--------correlation_package()
----README.md(2KB)
----example()
--------0img1.ppm(576KB)
--------0img0.ppm(576KB)
--------flow0.png(95KB)
--------1img1.ppm(576KB)
--------1img0.ppm(576KB)
----demo.py(1005B)
----misc()
--------demo.png(25KB)