1. 程式人生 > >手上只有一個模型檔案,怎樣用tensorboard檢視其模型圖Graph?

手上只有一個模型檔案,怎樣用tensorboard檢視其模型圖Graph?

比如,從他人處獲得一個Graph,想看看它的結構,怎麼弄?

Google提供了一個工具,TensorBoard,它能以圖表的方式分析你在訓練過程中彙總的各種資料,其中包括Graph結構。

所以我們可以簡單的寫幾行Pyhton,載入Graph,只在logdir裡,輸出Graph結構資料,並可以檢視其圖結構。

你可以在jupyter裡操作,程式碼如下:

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

# 這是從二進位制格式的pb檔案載入模型
graph = tf.get_default_graph()
graphdef = graph.as_graph_def()
graphdef.ParseFromString(gfile.FastGFile("/data/TensorFlowAndroidMNIST/app/src/main/expert-graph.pb", "rb").read())
_ = tf.import_graph_def(graphdef, name="")

#這是從檔案格式的meta檔案載入模型

_ = tf.train.import_meta_graph("model.ckpt.meta")

summary_write = tf.summary.FileWriter("/data/TensorFlowAndroidMNIST/logdir" , graph)

然後再啟動tensorboard:

tensorboard --logdir /data/TensorFlowAndroidMNIST/logdir --host 你的ip --port 你埠(預設6006)