tensorflow打印graph

今天药忘吃喽~ 2022-04-10 13:15 281阅读 0赞

参考：[https://blog.csdn.net/cooco369/article/details/82999172 ][https_blog.csdn.net_cooco369_article_details_82999172]

from tensorflow.python.framework import tensor_util
    from google.protobuf import text_format
    import tensorflow as tf
    from tensorflow.python.platform import gfile
    from tensorflow.python.framework import tensor_util
    
    
    GRAPH_PB_PATH = 'models/frozen_person_graph.pb' #path to your .pb file
    with tf.Session() as sess:
    	print("load graph")
    	with gfile.FastGFile(GRAPH_PB_PATH,'rb') as f:
    		graph_def = tf.GraphDef()
        # Note: one of the following two lines work if required libraries are available
    		#text_format.Merge(f.read(), graph_def)
    		graph_def.ParseFromString(f.read())
    		tf.import_graph_def(graph_def, name='')
    		for i,n in enumerate(graph_def.node):
    			print("Name of the node - %s" % n.name)

[https_blog.csdn.net_cooco369_article_details_82999172]: https://blog.csdn.net/cooco369/article/details/82999172%C2%A0

发表评论取消回复

表情：

评论列表（有 0 条评论，281人围观）

还没有评论，来说两句吧...

相关阅读

相关 tensorflow加载多graph

用tf加载不同网络结构的模型时，由于默认图的设置导致后边加载时可能出现节点不存在的错误提示，如下为先加载resnet的一个网络，然后加载vgg网络时提示的错误信息。 ...

落日映苍穹つ/ 2024年04月19日 14:53/ 0 赞/ 33 阅读

相关【Tensorflow】tf.Graph()函数

tf.Graph() 函数非常重要，注意提现在两个方面 > > * **它可以通过tensorboard用图形化界面展示出来流程结构** > * **它可...

小鱼儿/ 2024年04月18日 14:09/ 0 赞/ 33 阅读

相关【Tensorflow】tf.import_graph_def

tf.import\_graph\_def `tf.import_graph_def( graph_def, input_map=None, return_eleme...

小咪咪/ 2024年04月17日 16:46/ 0 赞/ 28 阅读

相关 tensorflow 出现ImportError: cannot import name graph_util

from tensorflow.python.framework import graph\_util 此句出现错误 ImportError: cannot impo

拼搏现实的明天。/ 2022年06月03日 07:39/ 0 赞/ 159 阅读

相关 tensorflow 获取graph中的所有tensor

[n.name for n in tf.get_default_graph().as_graph_def().node] 摘自[https://stackoverfl

我就是我/ 2022年05月19日 04:49/ 0 赞/ 179 阅读

相关 tensorflow统计graph中的trainable_variables

最简单的做法: 转自: [https://blog.csdn.net/feynman233/article/details/79187304][https_blog.csdn.

比眉伴天荒/ 2022年05月19日 03:34/ 0 赞/ 160 阅读

相关【tensorflow】Graph对象、Session对象、占位符

学习笔记TF005:Graph对象、Session对象、占位符 2017.05.10 03:07\ 字数 736 阅读 167 评论 0 喜欢 0 创建Graph对象，

拼搏现实的明天。/ 2022年05月18日 06:23/ 0 赞/ 150 阅读

相关 tensorflow打印graph

参考：[https://blog.csdn.net/cooco369/article/details/82999172 ][https_blog.csdn.net_cooco3

今天药忘吃喽~/ 2022年04月10日 13:15/ 0 赞/ 282 阅读

相关打印tensorflow结构

代码： import tensorflow as tf sess=tf.Session with tf.Graph().as_default

电玩女神/ 2022年02月02日 00:04/ 0 赞/ 201 阅读

相关 TensorFlow为什么要使用Graph？

主要有以下几点原因：（1）节省资源、高效运算我么只需要运算需要的依赖项，不需要的不进行运算。（2）把整个运算分解成子环节，方便自动求导；（3）对分布式运算友好，

Dear 丶/ 2021年12月19日 00:09/ 0 赞/ 203 阅读