pytorch-pwc:与官方Caffe版本匹配的PyTorch中PWC-Net的重新实现

pytorch-pwc 这是使用PyTorch对PWC-Net [1]的个人重新实现。 如果您要利用这项工作，请相应地引用本文。 另外，请确保遵守作者的。 如果您要使用此特定实现，请适当确认它[2]。 有关这项工作的原始版本，请参见： : 我的另一个光流实现： : 我的另一个光流实现： : 我的另一种光流实现： : 背景 幸运的是，PWC-Net的作者已经在PyTorch中提供了参考实现。 但是，其初始版本未达到原始Caffe版本的性能。 这就是为什么我创建此存储库的原因，在其中我通过利用其权重来复制Caffe官方版本的性能。 从那时起，正式的PyTorch实施就采用了我使用Caffe权重的方法，这就是为什么它们现在都表现出色的原因。 但是，许多人在尝试运行正式的PyTorch版本时都报告了CUDA的问题，而我的重新实现似乎不受此类问题的影响。 设置 相关层是使用Cu