【文件属性】：
文件名称：SCV:通过少量匹配学习光流（CVPR 2021）
文件大小：112.6MB
文件格式：ZIP
更新时间：2021-05-01 02:06:18
Python
通过少量匹配学习光流
该存储库包含本文的源代码：
CVPR 2021 江诗豪，卢瑶，李洪东，理查德·哈特利澳大利亚国立大学
要求
该代码已通过PyTorch 1.6和Cuda 10.1进行了测试。
conda create --name scv
conda activate scv
conda install pytorch=1.6.0 torchvision=0.7.0 cudatoolkit=10.1 matplotlib tensorboard scipy opencv -c pytorch
pip install faiss-gpu
所需数据
要评估/训练SCV，您将需要下载所需的数据集。
（可选）
默认情况下， datasets.py将在这些位置搜索数据集。 您可以创建符号链接，以链接到数据datasets文件夹中下载datasets
├── datasets
【文件预览】：
SCV-main
----chairs_split.txt(45KB)
----train.py(10KB)
----checkpoints()
--------quarter()
--------eighth()
----core()
--------utils()
--------sparsenet.py(13KB)
--------update.py(5KB)
--------datasets.py(9KB)
--------compute_sparse_correlation.py(2KB)
--------__init__.py(0B)
--------knn.py(3KB)
--------extractor.py(9KB)
----scv.png(230KB)
----train.sh(1KB)
----LICENSE(482B)
----README.md(2KB)
----evaluate.py(6KB)
----.gitignore(13B)
----evaluate_vis.py(14KB)