此项目提供了 PyTorch 框架下 TVNet 光流生成模型的实现。该实现简洁易懂,仅包含约 350 行代码,并遵循 PyTorch 模块化风格,方便扩展。最初的 TVNet 实现使用 TensorFlow 框架,可以在这里找到。此外,该项目还提供了演示代码,展示了如何使用 TVNet 生成光流表示,包括对流图的可视化。

要求:

  • Python 3
  • PyTorch
  • OpenCV(可选)

用法:

  1. 将输入帧放入 frame/img1.pngframe/img2.png
  2. 运行 python demo.py 生成光流表示。