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://github.com/ildoonet/tf-pose-estimation/issues/167
012020-03-31
相似问题