脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → tensorflow 打印pb模型节点

tensorflow 实现打印pb模型的所有节点

 更新时间：2020年01月23日 17:39:19   作者：coocoky  
今天小编就为大家分享一篇tensorflow 实现打印pb模型的所有节点，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

只有pd模型文件, 打印所有节点

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)

以上这篇tensorflow 实现打印pb模型的所有节点就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • 基于Python实现图片一键切割九宫格的工具

    基于Python实现图片一键切割九宫格的工具

    有时候发微博时候，需要裁切图片为九宫格，但是ps或者其他工具都太麻烦。本文就来用Python编写一个一键切割九宫格的工具，希望对大家有所帮助
    2023-03-03
  • 使用Python爬取弹出窗口信息的实例

    使用Python爬取弹出窗口信息的实例

    这篇文章主要介绍了使用Python爬取弹出窗口信息的实例，具有很好的的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-03-03
  • python numpy创造图像矩阵示例详解

    python numpy创造图像矩阵示例详解

    这篇文章主要为大家介绍了python numpy创造图像矩阵示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-08-08
  • 利用tkinter实现下拉框联动

    利用tkinter实现下拉框联动

    这篇文章主要介绍了利用tkinter实现下拉框联动问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
    2023-01-01
  • Python 图像处理之PIL库详解用法

    Python 图像处理之PIL库详解用法

    对于图像识别，大量的工作在于图像的处理，处理效果好，那么才能很好地识别，因此，良好的图像处理是识别的基础。在Python中，有一个优秀的图像处理框架，就是PIL库，本文会介绍PIL库中的各种方法，并列举相关例子
    2021-11-11
  • numpy.transpose()实现数组的转置例子

    numpy.transpose()实现数组的转置例子

    今天小编就为大家分享一篇numpy.transpose()实现数组的转置例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-12-12
  • python中for循环的多种使用实例

    python中for循环的多种使用实例

    for语句是Python中执行迭代的两个语句之一,另一个语句是while,下面这篇文章主要给大家介绍了关于python中for循环的多种使用方法,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
    2022-09-09
  • python清洗疫情历史数据的过程详解

    python清洗疫情历史数据的过程详解

    这篇文章主要介绍了python清洗疫情历史数据,包括数据获取方法及使用python读取csv的详细代码，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2022-05-05
  • Windows下Python使用Pandas模块操作Excel文件的教程

    Windows下Python使用Pandas模块操作Excel文件的教程

    Pandas是一个强大的Python数据分析模块,这里我们先使用ANACONDA来帮助获取Pandas所以来的一些环境,然后来初步学习Windows下Python使用Pandas模块操作Excel文件的教程
    2016-05-05
  • Python网络编程基于多线程实现多用户全双工聊天功能示例

    Python网络编程基于多线程实现多用户全双工聊天功能示例

    这篇文章主要介绍了Python网络编程基于多线程实现多用户全双工聊天功能,结合实例形式分析了Python网络编程中使用多线程进行多用户异步通信的原理与相关实现技巧,需要的朋友可以参考下
    2018-04-04

最新评论