[Tensorflow]-[技術]Tensorflow訓練好的模型的保存和讀取

pb格式：

保存為pb格式主要有兩種方式：

1. 第一種方式使用graph_util.convert_variables_to_constants()

from tensorflow.python.framework import graph_utilconstant_graph = graph_util.convert_variables_to_constants(sess,sess.graph_def,[graph的名字])with tf.gfile.FastGFile(model.pb,mode=wb) as f: f.write(constant_graph.SerializeToString())

2.第二種方式使用tf.get_default_graph().as_graph_def()

constant_graph = tf.get_default_graph().as_graph_def()with tf.gfile.FastGFile(model.pb,mode=wb) as f: f.write(constant_graph.SerializeToString())