pytorch-unflow:与官方TensorFlow版本相匹配的PyTorch中UnFlow的重新实现

pytorch-unflow 这是使用PyTorch对UnFlow [1]的个人重新实现。 如果您要利用这项工作，请相应地引用本文。 另外，请确保遵守作者的。 如果您要使用此特定实现，请适当确认它[2]。 有关这项工作的原始TensorFlow版本，请参阅： : 我的另一个光流实现： : 我的另一个光流实现： : 我的另一种光流实现： : 设置 相关层是使用CuPy在CUDA中实现的，这就是为什么CuPy是必需的依赖项的原因。 可以使用pip install cupy安装它，也可以使用CuPy存储库中概述的提供的之一进行pip install cupy 。 用法 要在自己的映像对上运行它，请使用以下命令。 您可以在两种型号之间进行选择，请确保查看其图纸/代码以获取更多详细信息。 python run.py --model css --first ./images/fi