发新帖

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

[复制链接]
835 3

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
在目标检测任务中,如何将保存的ckpt模型文件转为pb文件。目标检测网络为faster-rcnn。所用代码项目地址:https://github.com/endernewton/tf-faster-rcnn
我知道答案 回答被采纳将会获得10 金币 已有3人回答
本楼点评(0) 收起

精彩评论3

田利民  TF荚荚  发表于 2018-4-16 18:05:24 | 显示全部楼层
freeze graph
本楼点评(0) 收起
TianLin  TF荚荚  发表于 2018-4-17 10:31:31 | 显示全部楼层
比较建议在训练的过程中(或者训练结束时)就保存成SavedModel格式的pb。这样保存出来就是pb,也不用转了。

  1. tf.saved_model.simple_save(session,
  2.     export_dir,
  3.     inputs={"x": x, "y": y},
  4.     outputs={"z": z})
复制代码


https://tensorflow.google.cn/ver ... d_model/simple_save

如果需要转的话,大致流程是在内容中重建graph, session, 然后再用上面的函数保存一下。
本楼点评(0) 收起
tensorfyx  TF豆豆  发表于 2018-7-3 17:45:08 | 显示全部楼层
https://zhuanlan.zhihu.com/p/32887066

可以参考这个文章,讲解的很全了。
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

2

积分
快速回复 返回顶部 返回列表