文件名称:SCV:通过少量匹配学习光流(CVPR 2021)
文件大小:112.6MB
文件格式:ZIP
更新时间:2024-05-20 19:52:58
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)