TensorFlow：从模型文件看它的结构

meta文件读取方式

import tensorflow as tf

from tensorflow.python.platform import gfile #这是从文件格式的meta文件加载模型

graph = tf.get_default_graph()

graphdef = graph.as_graph_def()

_ = tf.train.import_meta_graph("***/model.meta")

summary_write = tf.summary.FileWriter("***/summary/" , graph)

 

pb文件读取方式

import tensorflow as tf

from tensorflow.python.platform import gfile #这是从文件格式的meta文件加载模型

graph = tf.get_default_graph()

graphdef = graph.as_graph_def()

graphdef.ParseFromString(gfile.FastGFile("/****/model.pb", "rb").read())

_ = tf.import_graph_def(graphdef, name="")

summary_write = tf.summary.FileWriter("****/summary3/" , graph)

 

然后再启动tensorboard：

tensorboard --logdir /data/TensorFlowAndroidMNIST/logdir --host 你的ip --port 你端口(默认6006)

tensorboard --logdir "***/summary/"

tensorboard --logdir "***/summary2/"