使用tensorflow显示pb模型的所有网络结点方式
更新时间：2020年01月23日 17:45:09 作者：南有木兮木不知
今天小编就为大家分享一篇使用tensorflow显示pb模型的所有网络结点方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
代码如下：
import tensorflow as tf
from tensorflow.python.platform import gfile
model = 'pb_path'
graph = tf.get_default_graph()
graph_def = graph.as_graph_def()
graph_def.ParseFromString(gfile.FastGFile(model, 'rb').read())
tf.import_graph_def(graph_def, name='graph')
summaryWriter = tf.summary.FileWriter('log/', graph)
使用步骤：
1.修改display_model_nodes.py里的model定义为自己的pb模型
2.python display_model_nodes.py会在log里生成out日志
3.tensorboard --logdir=log
且会在下方出现网页地址 TensorBoard 1.10.0 at http://oeasy:6006 (Press CTRL+C to quit)
4.在浏览器中打输入：http://oeasy:6006就能看到模型的各个节点了
结果如下：
以上这篇使用tensorflow显示pb模型的所有网络结点方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Linux下安装python3.6和第三方库的教程详解,需要的朋友可以参考下2018-11-11
这篇文章主要介绍了Python基于更相减损术实现求解最大公约数的方法,简单说明了更相减损术的概念、原理并结合Python实例形式分析了基于更相减损术实现求解最大公约数的相关操作技巧与注意事项,需要的朋友可以参考下2018-04-04
这篇文章主要为大家详细介绍了python实现自动解数独小程序，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-01-01
这篇文章主要为大家详细介绍了Python如何不调用三方收费接口，照样实现识别图片为文字的功能。文中的示例代码讲解详细，感兴趣的可以了解一下2022-08-08
这篇文章主要介绍了Python机器学习之决策树算法,较为详细的分析了实例详解机器学习中决策树算法的概念、原理及相关Python实现技巧,需要的朋友可以参考下2017-12-12
这篇文章主要给大家介绍了关于pandas plot各种绘图的相关资料,文中通过图文以及实例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2022-03-03
大家好，本篇文章主要讲的是Python各种类型装饰器详细介绍，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览2021-12-12
readlines()是Python文件对象的一个方法，用于读取文件中的所有行并将它们作为一个字符串列表返回，这篇文章主要介绍了python readlines函数解析,需要的朋友可以参考下2023-06-06
这篇文章主要为大家详细介绍了Python实现感知器模型、两层神经网络，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2017-12-12
这篇文章主要介绍了Python如何获取excel数据，本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下2021-10-10
最新评论