tensorflow打印graph

今天药忘吃喽~ 2022-04-10 13:15 307阅读 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 条评论，307人围观）

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

相关阅读

相关【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 赞/ 60 阅读

相关 module 'tensorflow' has no attribute 'get_default_graph'

参考：https://blog.csdn.net/u014466109/article/details/88877321?depth\_1-utm\_source=distri

末蓝、/ 2023年07月15日 13:06/ 0 赞/ 42 阅读

相关 tensorflow生成常量_TensorFlow 之计算图Graph的构建和执行

计算图(Graph)用来表示计算任务。图中的节点称为op(operation)。计算图必须在会话中启动，会话会把op分发到cpu或gpu等计算设备上进行执行，执行后会返

￡神魔★判官ぃ/ 2023年01月10日 05:09/ 0 赞/ 146 阅读

相关 tensorflow 出现ImportError: cannot import name graph_util

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

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

相关 tensorflow 获取graph中的所有tensor

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

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

相关 tensorflow统计graph中的trainable_variables

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

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

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

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

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

相关 tensorflow打印graph

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

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

相关打印tensorflow结构

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

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

相关 TensorFlow为什么要使用Graph？

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

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