ckpt的模型怎么转换为pb格式的模型

来源:10-30 模型预测实现

慕粉3575308

2020-03-23

请问tf.estimator.Estimator()保存的.ckpt的模型怎么转换为pb格式的模型,因为生产环境要用TensorFlow Serving来做预测,而.ckpt格式的不支持tf Serving这种方式,在线等,挺急的!图片描述
.ckpt格式为上图

写回答

1回答

正十七

2020-03-30

不管checkpoint是怎么产生的,都可以使用freeze工具转成pb,可以参考

https://stackoverflow.com/questions/45864363/tensorflow-how-to-convert-meta-data-and-index-model-files-into-one-graph-pb

https://github.com/ildoonet/tf-pose-estimation/issues/167

0
1
慕粉3575308
多谢老师,我尝试一下!有问题再麻烦老师。
2020-03-31
共1条回复

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程