比如在神经网络做预测的时候,只需要保存部分的节点,比如变量的初始化这些就不需要保存。
比如:
import tensorflow as tf;
import numpy as np;
import matplotlib.pyplot as plt;
from tensorflow.python.framework import graph_util
v1 = tf.Variable(tf.constant(1.0, shape=[1]), name='v1')
v2 = tf.Variable(tf.constant(2.0, shape=[1]), name='v2')
result = v1 + v2
init = tf.initialize_all_variables()
with tf.Session() as sess:
sess.run(init)
graph_def = tf.get_default_graph().as_graph_def()
output_graph_def = graph_util.convert_variables_to_constants(sess, graph_def, ['add'])
with tf.gfile.GFile('/home/penglu/Desktop/lplp/model.pb', "wb") as f:
f.write(output_graph_def.SerializeToString())
导入模型:
import tensorflow as tf;
import numpy as np;
import matplotlib.pyplot as plt;
from tensorflow.