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

在目标检测任务中,如何将保存的 ckpt 模型文件转为 pb 文件。目标检测网络为 faster-rcnn。所用代码项目地址:https://github.com/endernewton/tf-faster-rcnn


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

freeze graph


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

比较建议在训练的过程中(或者训练结束时)就保存成 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

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


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