【RAFT】运行demo.py得到RGB图相应的光流图

RAFT的Github项目链接： https://github.com/princeton-vl/RAFT
相关论文：RAFT: Recurrent All Pairs Field Transforms for Optical Flow

运行RAFT的demo

安装库包

这里需要注意的是不要照着READEME.md安装相应库包的版本，亲测失败=.=
正确的做法是先访问pytorch官网https://pytorch.org/get-started/locally/，这个官网会根据你当前的主机情况，生成安装pytorch、torchvision等库包的命令行，然后复制到自己的conda虚拟环境下进行安装
最后安装matplotlib、tensorboard、scipy、opencv-python即可。opencv-python在安装过程中可能会因网络问题而中断，这时只需添加清华源即可解决

//笔者的主机显卡是RTX 4060ti
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
conda install matplotlib
conda install tensorboard
conda install scipy
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple

下载预训练模型

README.md给出两个方法，一个是运行命令行，另一个是从google drivehttps://drive.google.com/drive/folders/1sWDsfuZ3Up38EUQt7-JDTT1HcGHuJgvT下载，笔者采用后者方法，下载后解压到项目文件夹下。

运行demo.py

由于项目已经过四年了，demo.py上有的地方需要做调整：

//原来是这样，直接运行的话会提示不存在库包raft、utils
from raft import RAFT
from utils import flow_viz
from utils.utils import InputPadder
//经检查发现raft和utils是作者自定义的库包，存放在core文件夹下，因此代码应更改如下
from core.raft import RAFT
from core.utils import flow_viz
from core.utils.utils import InputPadder

保存demo.py，运行以下命令行：

//demo-frames是作者给出的示例图片文件夹
python demo.py --model=models/raft-things.pth --path=demo-frames

结果如下：
demo.py没有自动保存图片的功能，这需要自己添加~