1.用tensorflow框架,安装时注意cuda与tensorflow的对应关系,github上写明1.10.0,对应的cuda为9.0.
2.下载model文件,解压,生成对应的.mata index和data(tensorflow更新后,将ckpt分解为几个文件)
旧版保存单个模型文件的方式,大同小异,可以看看。
- .ckpt文件是旧版本的输出
saver.save(sess)
,相当于你的.ckpt-data
(见下文) - “checkpoint”文件仅用于告知某些TF函数,这是最新的检查点文件。
新版保存为:
- .ckpt.meta:包含元图,即计算图的结构,没有变量的值
- .ckpt.index:.data文件中数据和 .meta文件中结构图之间的对应关系(也就是变量的名称)
- .ckpt.data:保存的是变量值
网上参考的很多,比如https://zhuanlan.zhihu.com/p/53814653
3.需要修改dataset和model的路径:
4.test时运行:
python run_test.py --cfg==./configurations/kitti.json
在对应的result/DWARF/KITTI下生成预测结果图。