tfoptflow

收录理由

光流估计用于生成视频帧间的密集运动场，在目标跟踪、动作识别等任务中很常见。这个仓库是 PWC-Net 的 TensorFlow 实现，PWC-Net 是 CVPR 2018 提出的架构，代码覆盖了训练、评估和推理的完整流程，而不只是提供一个冻结模型。作者特意没有使用动态加载的 CUDA 算子，因此能在 Linux 和 Windows 上运行，也支持多 GPU 和混合精度训练，方便你用自己的数据重新训练。仓库还提供了预训练权重，在 MPI-Sintel 'final' 基准上的结果与原始论文相当或更优。如果你想找一个经典光流网络的参考实现来在此基础上开发，这是个不错的起点。