查看.pb文件的结构

最新推荐文章于 2024-07-05 14:08:57 发布

原创最新推荐文章于 2024-07-05 14:08:57 发布 · 1w 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#tensorboard #python

Tensorflow 专栏收录该内容

9 篇文章

订阅专栏

本文介绍如何使用TensorFlow将训练完成的model.pb文件导入到TensorBoard中，以便于直观地展示其网络结构。具体步骤包括：从PB文件恢复计算图、运行代码生成events文件以及通过TensorBoard加载并查看网络结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

1. 从pb文件中恢复计算图

import tensorflow as tf

model = 'model.pb'   #请将这里的model.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文件