将模型.pb文件在tensorboard中展示结构

小程序：扫一扫查出行
【扫一扫了解最新限行尾号】
复制小程序

本文介绍将训练好的model.pb文件在tensorboard中展示其网络结构。

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/events.out.tfevents.1535079670.DESKTOP-5IRM000。

2. 在tensorboard中加载