博客

我不是知识的生产者，我只是一个渺小的搬运工，我们都站在巨人的肩膀上

探索了一下午这玩意的用法，终于会用了，在此附上实例子

首先要明白他保存图的原理，这个里面讲的很详细，请细品

https://zhuanlan.zhihu.com/p/31308381

tf.train.write_graph这个函数可以保留节点，op，constant，但不保存variable，如果你想要保存variable，那么就要转为constant

import tensorflow as tf
import numpy as np
from tensorflow.python.platform import gfile

#生成图
input1= tf.placeholder(tf.int32,name="input")
b = tf.constant([3])
output1= tf.add(input1, b, name="output")

#保存图
with tf.Session() as sess:
    tf.train.write_graph(sess.graph_def, "./", "test.pb", False)
    print(sess.run(output1,feed_dict={input1:1}))

#读取图
with tf.Session() as sess:
    with gfile.FastGFile("./test.pb",'rb') as f:
        graph = tf.get_default_graph()
        graph_def = graph.as_g