步骤1:从pb文件中恢复计算图
import tensorflow as tf
model = '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) #log存放地址
步骤2:tensorboard查看计算图
命令行中输入
tensorboard --logdir log/ #log/ 为log存放地址