如何使用tensorboard查看tensorflow 　graph****.pb文件的模型结构

转载于 2018-12-19 14:38:00 发布 · 322 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/YouXiangLiThon/p/10143106.html

文章标签：

#人工智能 #c#

本文介绍了如何使用TensorFlow加载.pb格式的模型文件，并通过TensorBoard进行模型结构的可视化展示。具体步骤包括在Session中导入模型，使用FastGFile读取.pb文件，解析并导入GraphDef，最后使用summary.FileWriter将模型添加到指定的日志目录下。

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

参考网上的：https://github.com/tensorflow/tensorflow/issues/8854

import tensorflow as tf
from tensorflow.python.platform import gfile
with tf.Session() as sess:
    model_filename ='PATH_TO_PB.pb'
    with gfile.FastGFile(model_filename, 'rb') as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        g_in = tf.import_graph_def(graph_def)
　　　　　LOGDIR='YOUR_LOG_LOCATION'
　　　　　train_writer = tf.summary.FileWriter(LOGDIR)
　　　　　train_writer.add_graph(sess.graph)

转载于:https://www.cnblogs.com/YouXiangLiThon/p/10143106.html