查看TensorFlow的pb模型文件的ops和tensor并使用TensorBoard可视化

本文介绍了如何加载.pb模型文件,查看其ops和tensor,并利用TensorBoard进行模型结构的可视化。通过加载模型,可以输出定义和tensor的shape。同时,详细解释了使用TensorBoard进行模型可视化的过程,包括在终端启动TensorBoard和在浏览器中查看网络结构。
摘要由CSDN通过智能技术生成
加载pb模型文件,并输出定义
model = 'model.pb'
with tf.Session() as sess:
    with open(model, 'rb') as model_file:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        print(graph_def)

采用上述的方式可以在新的会话中重新加载本地的模型文件(pb),然后二进制解析后,输出可以看到结果。但是如果网络层结构十分复杂,那么这种显示方式就会比较难以阅读。

加载pb模型文件,并使用Tensorboard可视化
import tensorflow 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值