RAFT 源码

筏 该存储库包含本文的源代码: ECCV 2020扎卡里·泰德(Jachary Teed)和贾登(Jia Deng) 要求 该代码已通过PyTorch 1.6和Cuda 10.1进行了测试。 conda create --name raft conda activate raft conda install pytorch=1.6.0 torchvision=0.7.0 cudatoolkit=10.1 matplotlib tensorboard scipy opencv -c pytorch 演示版 可以通过运行下载预训练的模型 ./download_models.sh 或从下载 您可以在一系列框架上演示训练有素的模型 python demo.py --model=models/raft-things.pth --path=demo-frames 所需数据 要评估/训练RAFT,您