FlowNetPytorch:Dosovitskiy等人的FlowNet的Pytorch实现

时间:2021-05-24 13:04:00
【文件属性】:
文件名称:FlowNetPytorch:Dosovitskiy等人的FlowNet的Pytorch实现
文件大小:1.1MB
文件格式:ZIP
更新时间:2021-05-24 13:04:00
flow pytorch flownet Python FlowNetPytorch Dosovitskiy等人的FlowNet的Pytorch实现。 该存储库是等人的的火炬实现。 在PyTorch中。 查看Torch的实现 这段代码主要是受官方启发。 尚未针对多个GPU进行过测试,但是它应该与原始代码一样工作。 该代码使用提供了训练示例,并进行数据扩充。 将来可能会添加的实现。 当前提供了两个神经网络模型,以及它们的批处理规范变化(实验性): FlowNetS FlowNetSBN FlowNetC FlowNetCBN 预训练模型 由于您可以(从pytorch从朱古力,不)下载FlowNetS的预训练版本。 此文件夹还包含从头开始训练有素的网络。 有关网络加载的注意事项 将下载的网络直接输入脚本,即使您的桌面环境告诉您,也无需解压缩它。 注意来自caffe的网络 这些网络需要BGR输入(与pytorch中的RGB相比)。 但
【文件预览】:
FlowNetPytorch-master
----.gitignore(19B)
----images()
--------GT_3.png(7KB)
--------input_2.gif(353KB)
--------input_3.gif(275KB)
--------GT_1.png(18KB)
--------pred_2.png(13KB)
--------GT_2.png(16KB)
--------pred_3.png(10KB)
--------pred_1.png(13KB)
--------input_1.gif(405KB)
----requirements.txt(109B)
----datasets()
--------mpisintel.py(3KB)
--------__init__.py(261B)
--------util.py(809B)
--------KITTI.py(4KB)
--------flyingchairs.py(1KB)
--------listdataset.py(2KB)
----flow_transforms.py(9KB)
----models()
--------FlowNetC.py(5KB)
--------__init__.py(47B)
--------FlowNetS.py(5KB)
--------util.py(2KB)
----LICENSE(1KB)
----README.md(6KB)
----util.py(1KB)
----main.py(13KB)
----multiscaleloss.py(2KB)
----run_inference.py(5KB)

网友评论