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

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

转自: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)


您可能感兴趣的与本文相关的镜像

TensorFlow-v2.15

TensorFlow-v2.15

TensorFlow

TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值