pb文件使用tensorboard查看網絡結構

親測可用

轉載：https://www.jianshu.com/p/8ad8a401f196

新建generate_graph.py文件

import os
import tensorflow as tf

model = '/mnt/e/model.pb' #請將這裏的pb文件路徑改爲自己的
graph = tf.get_default_graph()
graph_def = graph.as_graph_def()
graph_def.ParseFromString(tf.gfile.FastGFile(model, 'rb').read())
tf.import_graph_def(graph_def, name='graph')
summaryWriter = tf.summary.FileWriter('log/', graph)

運行python generate_graph.py，此時會在log目錄下生成一個log文件。

tensorboard --logdir log

此時會出現
TensorBoard 1.13.1 at http://127.0.0.1:6006 (Press CTRL+C to quit)
打開瀏覽器，輸入http://127.0.0.1:6006，回車，可以看到網絡結構。