此项目提供了 PyTorch 框架下 TVNet 光流生成模型的实现。该实现简洁易懂,仅包含约 350 行代码,并遵循 PyTorch 模块化风格,方便扩展。最初的 TVNet 实现使用 TensorFlow 框架,可以在这里找到。此外,该项目还提供了演示代码,展示了如何使用 TVNet 生成光流表示,包括对流图的可视化。
要求:
- Python 3
- PyTorch
- OpenCV(可选)
用法:
- 将输入帧放入
frame/img1.png
和frame/img2.png
。 - 运行
python demo.py
生成光流表示。