新建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，回车，可以看到网络结构。
网友评论