pytorch-liteflownet:与官方Caffe版本相匹配的PyTorch中LiteFlowNet的重新实现

时间:2024-05-23 19:38:34
【文件属性】:

文件名称:pytorch-liteflownet:与官方Caffe版本相匹配的PyTorch中LiteFlowNet的重新实现

文件大小:1.69MB

文件格式:ZIP

更新时间:2024-05-23 19:38:34

python deep-learning cuda pytorch cupy

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


【文件预览】:
pytorch-liteflownet-master
----images()
--------first.png(529KB)
--------second.png(531KB)
--------README.md(85B)
----LICENSE(34KB)
----correlation()
--------correlation.py(14KB)
--------README.md(435B)
----run.py(17KB)
----comparison()
--------official - caffe.png(152KB)
--------comparison.py(1KB)
--------this - pytorch.png(152KB)
--------comparison.gif(344KB)
----requirements.txt(52B)
----README.md(3KB)

网友评论