标签: android python tensorflow
Android Tensorflow Demo中的初始网络将模型存储为protobuf文件(tensorflow_inception_graph.pb)。我想用另一个网络替换这个网络。
是否有关于如何将Python中训练有素的TensorFlow网络转换为.pb的文档或示例?我知道TensorFlow的Saver,但这似乎用于保存中级训练状态。如果模型已经过培训,不确定它是如何工作的。
.pb
答案 0 :(得分:5)
以下是saving和loading
答案 1 :(得分:4)
tf.Graph.as_graph_def()检索序列化图表。然后你就把它写到一个文件中。
tf.Graph.as_graph_def()