flownet2.tf:TensorFlow-1.2.0中的现成FlowNet模块

时间:2024-05-28 18:14:20
【文件属性】:

文件名称:flownet2.tf:TensorFlow-1.2.0中的现成FlowNet模块

文件大小:1.73MB

文件格式:ZIP

更新时间:2024-05-28 18:14:20

optical-flow C++

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 如果安装成功,


【文件预览】:
flownet2.tf-master
----demo.py(1KB)
----LICENSE(1KB)
----.gitignore(93B)
----FlowNet2_src()
--------training_schedules.py(301B)
--------flownet_css()
--------dataset_configs.py(4KB)
--------flownet2()
--------example()
--------flownet_cs()
--------flowlib.py(14KB)
--------utils.py(2KB)
--------dataloader.py(14KB)
--------flownet_s()
--------flownet_sd()
--------__init__.py(268B)
--------ops()
--------correlation.py(2KB)
--------checkpoints()
--------net.py(6KB)
--------flow_warp.py(477B)
--------downsample.py(213B)
--------flownet_c()
----Makefile(4KB)
----README.md(2KB)

网友评论