保存tensorflow模型为pb文件

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

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

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

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

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

参考链接： https://www.jianshu.com/p/091415b114e2

https://cloud.tencent.com/developer/ask/188650

由于arm nn官方提供的mnist-tf例程中提供的模型类型是prototxt或者pb文件，所以这里需要把tensorflow保存的ckpt文件转换成pb文件

tensorflow训练生成的ckpt文件包含4个，分别是

[原文连接]（ https://blog.csdn.net/u014061630/article/details/80557028 ）

方法一：

方法二

在命令行运行以下命令，启动Tensorboard

或者

经过查看源码，第二种方法其实是对第一种方法的包装。

两种方法是一致的，只不过第二种方法更加便捷。

---