在目标检测任务中，如何将保存的 ckpt 模型文件转为 pb 文件

技术问答
1

在目标检测任务中，如何将保存的 ckpt 模型文件转为 pb 文件。目标检测网络为 faster-rcnn。所用代码项目地址：GitHub - endernewton/tf-faster-rcnn: Tensorflow Faster RCNN for Object Detection

提问人： 少奇 a，发帖时间：2018-4-16 15:20:30

2

freeze graph

田利民，发表于 2018-4-16 18:05:24

3

比较建议在训练的过程中（或者训练结束时）就保存成 SavedModel 格式的 pb。这样保存出来就是 pb，也不用转了。

tf.saved_model.simple_save (session,
    export_dir,
    inputs={"x": x, "y": y},
    outputs={"z": z})

https://tensorflow.google.cn/versions/master/api_docs/python/tf/saved_model/simple_save

如果需要转的话，大致流程是在内容中重建 graph, session, 然后再用上面的函数保存一下。

TianLin，2018-4-17 10:31:31

4

可以参考这个文章，讲解的很全了。

tensorfyx，发表于 2018-7-3 17:45:08