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/"

 

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章