文件名称：视频图matlab代码-pytorch-tvnet:TVNet在pytorch中生成光流
文件大小：1.57MB
文件格式：ZIP
更新时间：2024-06-16 08:15:54
系统开源
视频图matlab代码pytorch电视网 该项目包含pytorch-style的TVNet的简单和pytorch实现。 简单：总共约350行代码 Pytorch样式：所有模块（*渐变，正向渐变和散度）都扩展为torch.nn.Module。 最初的实现是在tensorflow中，可以在中找到。 更新2020/04/20 ：GPU支持，请尝试demo_gpu.py; 支持批处理模式，请尝试demo_batch_mode.py。 要求 Python 3 ：也在python 2.7上进行了测试。 火炬 matlab（optinonal） ：在原始的tensorflow版本中，作者使用.mat文件保存TVNet生成的结果，使用Matlab保存结果。 在演示代码中，我还添加了用于使用cv2（在python中）对流图进行可视化的代码。 用法 I）将输入帧放在frame/img1.png ， frame/img2.png 。 II）使用TVNet生成运动表示 用法示例包括 为frame/img1.png和frame/img2.png中的frame/img1.png生成运动表示。 python d
【文件预览】：
pytorch-tvnet-master
----tvnet.py(14KB)
----visualize()
--------result-pytorch.png(72KB)
--------visualize.m(143B)
--------result.png(72KB)
--------computeColor.m(3KB)
--------readFlowFile.m(3KB)
--------flow-pytorch.png(75KB)
--------colorTest.m(2KB)
--------README.txt(298B)
--------writeFlowFile.m(2KB)
--------flowToColor.m(3KB)
--------pytorch-result.png(72KB)
----demo_batch_mode.py(2KB)
----frame()
--------img1.png(115KB)
--------img2.png(116KB)
----LICENSE(1KB)
----demo.py(2KB)
----README.md(2KB)
----demo_gpu.py(2KB)
----result()
--------result-pytorch.png(72KB)
--------result.png(72KB)
--------result-pytorch.mat(900KB)
--------result.mat(900KB)