保存tensorflow模型为pb文件

Posted 2023-05-15

参考技术A 通常训练模型的时候是保存ckpt方便接着训练，但是上线可以保存为pb模型，加载的时候不需要重新定义模型，只用输入输出来调用模型。

save_model 和load_model两个函数要分开执行，第一次注释掉load，只save，第二次load的时候注释掉save。因为声明模型的时候都是用默认图，变量命名会依次是0,1，load的时候名字对应不上。

保存好的pb模型路径文件格式为

还有一种保存pb模型的方法，保存的东西只有model.pb没有variables。

如果需要发布模型成服务，请看： https://www.jianshu.com/p/5b74f1bc0178

如何用tensorflow训练模型成pb文件和和如何加载已经训练好的模型文件

参考技术A 基本使用 使用 TensorFlow, 你必须明白 TensorFlow: 使用图 (graph) 来表示计算任务. 在被称之为 会话 (Session) 的上下文 (context) 中执行图. 使用 tensor 表示数据. 通过 变量 (Variable) 维护状态. 使用 feed 和 fetch 可以为任意的操作(arb...