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

时间:2016-02-06 22:41:35

标签: android python tensorflow

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

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

2 个答案:

答案 0 :(得分:5)

以下是savingloading

的示例

答案 1 :(得分:4)

tf.Graph.as_graph_def()检索序列化图表。然后你就把它写到一个文件中。

相关问题