【Tensorflow】Tensorboard可视化查看tensorflow pb模型节点

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

model = 'models/ssd_mobilenet_v2_coco_2018_03_29/frozen_inference_graph.pb'
graph = tf.get_default_graph()
graph_def = graph.as_graph_def()
graph_def.ParseFromString(gfile.FastGFile(model, 'rb').read())
tf.import_graph_def(graph_def, name='graph')
summaryWriter = tf.summary.FileWriter('log/', graph)

在当前目录下生成了一个log文件夹，里面有一个events文件。

打开anaconda prompt，在终端输入

tensorboard --logdir=log

将http://LAPTOP-QCL6LUUM:6006复制到浏览器打开。