文件名称：pytorch-pwc:与官方Caffe版本匹配的PyTorch中PWC-Net的重新实现
文件大小：1.65MB
文件格式：ZIP
更新时间：2024-05-23 19:34:01
python deep-learning cuda pytorch cupy
pytorch-pwc 这是使用PyTorch对PWC-Net [1]的个人重新实现。 如果您要利用这项工作，请相应地引用本文。 另外，请确保遵守作者的。 如果您要使用此特定实现，请适当确认它[2]。 有关这项工作的原始版本，请参见： : 我的另一个光流实现： : 我的另一个光流实现： : 我的另一种光流实现： : 背景 幸运的是，PWC-Net的作者已经在PyTorch中提供了参考实现。 但是，其初始版本未达到原始Caffe版本的性能。 这就是为什么我创建此存储库的原因，在其中我通过利用其权重来复制Caffe官方版本的性能。 从那时起，正式的PyTorch实施就采用了我使用Caffe权重的方法，这就是为什么它们现在都表现出色的原因。 但是，许多人在尝试运行正式的PyTorch版本时都报告了CUDA的问题，而我的重新实现似乎不受此类问题的影响。 设置 相关层是使用Cu
【文件预览】：
pytorch-pwc-master
----images()
--------second.png(531KB)
--------README.md(85B)
--------first.png(529KB)
----requirements.txt(52B)
----run.py(14KB)
----LICENSE(34KB)
----README.md(3KB)
----correlation()
--------correlation.py(13KB)
--------README.md(435B)
----comparison()
--------this - pytorch.png(132KB)
--------official - caffe.png(131KB)
--------comparison.py(1KB)
--------comparison.gif(340KB)