将训练有素的Tensorflow模型转换为protobuf

Question

Android Tensorflow演示中的初始网络将模型存储为protobuf文件(tensorflow_inception_graph.pb).我想用另一个网络替换这个网络.

是否有文档或示例如何将Python中训练有素的TensorFlow网络转换为.pb？我知道TensorFlow的Saver,但这似乎用于保存中级训练状态.如果模型已经过培训,不确定它是如何工作的.

Answer 1

这是保存和加载的示例

Answer 2

tf.Graph.as_graph_def()检索序列化图。然后你只需将其写入文件即可。