手上只有一个模型文件,怎样用tensorboard查看其模型图Graph?

本文介绍如何利用TensorBoard工具来展示和分析TensorFlow中Graph的结构。通过几行Python代码即可实现Graph结构数据的输出,并能在指定目录下查看到Graph的可视化结构。文中提供了从.pb或.meta文件加载模型的具体步骤。

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

转自:https://blog.youkuaiyun.com/czq7511/article/details/72480149

比如,从他人处获得一个Graph,想看看它的结构,怎么弄?

Google提供了一个工具,TensorBoard,它能以图表的方式分析你在训练过程中汇总的各种数据,其中包括Graph结构。

所以我们可以简单的写几行Pyhton,加载Graph,只在logdir里,输出Graph结构数据,并可以查看其图结构。

可参考:http://www.tensorfly.cn/tfdoc/how_tos/summaries_and_tensorboard.html

https://www.tensorflow.org/get_started/summaries_and_tensorboard

你可以在jupyter里操作,代码如下:

import tensorflow as tf
from tensorflow.python.platform import gfile

# 这是从二进制格式的pb文件加载模型
graph = tf.get_default_graph()
graphdef = graph.as_graph_def()
graphdef.ParseFromString(gfile.FastGFile("/data/TensorFlowAndroidMNIST/app/src/main/expert-graph.pb", "rb").read())
_ = tf.import_graph_def(graphdef, name="")


#这是从文件格式的meta文件加载模型

_ = tf.train.import_meta_graph("model.ckpt.meta")


summary_write = tf.summary.FileWriter("/data/TensorFlowAndroidMNIST/logdir" , graph)


然后再启动tensorboard:

tensorboard --logdir /data/TensorFlowAndroidMNIST/logdir --host 你的ip --port 你端口(默认6006)


### TensorBoard 图片显示不全的解决方案 在使用 TensorBoard 进行可视化时,如果遇到图片显示不全的问题,可以尝试以下方法来解决问题。 1. **检查日志路径是否正确** 确保在启动 TensorBoard 时,`--logdir` 参数指定的日志路径是正确的。如果使用相对路径导致问题,可以改为绝对路径。例如,将命令改为: ```bash tensorboard --logdir=C:\Users\Qian\Desktop\logs ``` 这样可以避免路径解析错误导致的图片加载失败[^1]。 2. **更换浏览器** 如果在默认浏览器(如 Edge)中打开 TensorBoard 的页面时出现乱码或显示异常,建议更换为其他浏览器(如 Chrome 或 Firefox)。此外,也可以尝试将地址改为 `localhost:6006` 来访问页面[^2]。 3. **清理多余的事件文件** 在当前路径下可能存在多个事件文件(`events.out.tfevents.*`),这可能会导致 TensorBoard 加载数据时出现问题。可以手动删除多余的事件文件,或者在代码中添加以下语句以重置默认图结构: ```python tf.reset_default_graph() ``` 这样可以确保每次运行时生成新的事件文件[^3]。 4. **调整采样参数** 如果 TensorBoard 显示的数据不全,可能是由于默认采样数量限制所致。可以通过增加采样数量来解决此问题。例如,在启动 TensorBoard 时添加参数 `--samples_per_plugin=scalars=100000`: ```bash tensorboard --logdir=logs --samples_per_plugin=scalars=100000 ``` 这会显著提高数据的采样率,从而减少显示不全的情况[^5]。 5. **更改端口避免冲突** 如果在同一台主机上有多人同时训练模型,可能会导致端口冲突或显示异常。可以通过指定不同的端口号来避免此类问题。例如: ```bash tensorboard --logdir=logs --port=6007 ``` 这样可以确保每个人使用的 TensorBoard 实例不会相互干扰[^4]。 通过以上方法,可以有效解决 TensorBoard图片显示不全的问题。 ```python # 示例代码:确保每次运行生成新的事件文件 import tensorflow as tf # 清空默认图结构 tf.reset_default_graph() # 定义简单的计算图 with tf.compat.v1.Session() as sess: writer = tf.compat.v1.summary.FileWriter("logs", sess.graph) writer.close() ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值