
Flownet2.TF:基于TensorFlow-1.2.0的预构建FlowNet模块
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
Flownet2.TF是基于TensorFlow-1.2.0开发的一个预构建FlowNet模块,用于实现光流估计。它提供了高效且灵活的框架以进行深度学习研究与应用。
FlowNet2(TensorFlow)
此回购包括FlowNetC、S、CS、CSS、CSS-ft-sd、SD 和 2 (用于 TensorFlow)。大部分内容来自相关资源,并且我们进行了一些调整:
现在可以处理任意大小的输入。
安装后,只需将整个文件夹 FlowNet2_src 复制到您的代码库即可使用。有关详细信息,请参见 demo.py 文件。
环境
该代码已经过 Python3.6 和 TensorFlow1.2.0 以及 Tesla K80 GPU 的测试。系统是 Ubuntu 14.04。
安装
您必须先安装 CUDA:make all。
注意:根据使用的 GPU,可能需要修改相关设置。
下载砝码
要下载所有模型的权重(约 4.4GB),请在 FlowNet2_src/checkpoints 目录中运行 download.sh 脚本。所有测试脚本都依赖于这些检查点来正常工作。
推论模式
python demo.py
全部评论 (0)
还没有任何评论哟~


