Tensorflow之pb文件分析

最新推荐文章于 2024-07-21 07:15:00 发布

黑暗星球

最新推荐文章于 2024-07-21 07:15:00 发布

阅读量1.5w

点赞数 4

分类专栏： TensorFlow教程文章标签： pb 模型

本文链接：https://blog.csdn.net/u014061630/article/details/80557028

版权

本文介绍了两种使用Tensorboard分析Tensorflow pb模型文件的方法：一是通过恢复计算图并利用Tensorboard查看；二是使用tensorflow tools的import_pb_to_tensorboard.py工具。尽管Linux版Tensorflow不默认包含此工具，但可以通过源码获取。两种方法实质相同，第二种方法更为便捷。

摘要由CSDN通过智能技术生成

使用Tensorboard分析pb文件有两种方法：

方法一：

利用pb文件恢复计算图
利用Tensorboard查看计算图的结构

方法二

利用tensorflow提供的tools里的import_pb_to_tensorboard.py这个工具，但是这个工具linux版本的tensorflow没有安装（Win下默认安装），需要的可以去下载[https://github.com/tensorflow/tensorflow/tree/master/tensorflow/python/tools]

方法一

1.从pb文件中恢复计算图

import tensorflow as tf

model = 'model.pb' #请将这里的pb文件路径改为自己的
graph = tf.get_default_graph()
graph_

最低0.47元/天解锁文章

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

黑暗星球

关注关注

4
点赞
踩
22

收藏

觉得还不错? 一键收藏
8
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

tensorflow保存数据为.ｐｂ格式和加载ｐｂ文件

fu6543210的博客

05-16

6万+

最近接触了ｔｅｎｓｏｒｆｌｏｗ的ｏｂｊｅｃｔ　ｄｅｔｅｃｔｉｏｎ　ＡＰＩ发现里面读取的预先训练模型都是ｐｂ格式。谷歌推荐的保存模型的方式是保存模型为 PB 文件，它具有语言独立性，可独立运行，封闭的序列化格式，任何语言都可以解析它，它允许其他语言和深度学习框架读取、继续训练和迁移 TensorFlow 的模型。它的主要使用场景是实现创建模型与使用模型的解耦，使得前向推导 inference的代码...

tensorflow从pb模型文件提取features

10-28

用pb文件，输入input，修改output_node,可以输出神经网络每层的输出features

8 条评论您还未登录，请先登录后发表或查看评论

SpringBoot2实战：TensorFlow模型加载指南——.pb文件解析

DOUZHAOPENG1113的博客

06-09

157

.pb模型文件具有语言独立性，可独立运行，封闭的序列化格式，可使用任何语言解析。*.pb模型文件中变量是固定的(const)即模型中的变量值固定存储。*.pb模型文件使用过程中不会重新“学习”，即模型参数不变，保证了模型的稳定性。*.pb模型文件实现了庞大模型的瘦身，即该格式的模型尺寸较小主要用于移动端。基础阅读:[1](一)Tensorflow搭建神经网络[2](二)Tensorflow神经网络保存模型（持久化）[3](三)Tensorflow神经网络之模型载入及迁移学习[参考文献]

PB （protobuf）学习以及项目应用梳理：

最新发布

weixin_62946182的博客

07-21

880

protobuf不支持rpc通信功能，只支持对于rpc方法的描述，通过这个描述它就可以去做rpc请求的携带的参数的序列化和反序列化，login的(LoginRequest)传输调用的方法名及参数，returns就是表示：这个rpc方法login(LoginResponse)执行以后返回的protobuf打包的类型。

tensorflow读取pb文件，出现的ValueError问题

m0_38064529的博客

05-26

1105

先记录，解决后来补充 ValueError: Input 0 of node import/MobilenetV2/expanded_conv/depthwise/BatchNorm/cond/Assign/Switch was passed float from import/MobilenetV2/expanded_conv/depthwise/BatchNorm/moving_mean:0 incompatible with expected float_ref. ...

tensorflow读取pb文件推理

pengpengloveqiaoqiao的博客

11-30

2682

import tensorflow as tf from tensorflow.python.platform import gfile import numpy as np import time (x_train, y_train),(x_test, y_test) = tf.keras.datasets.mnist.load_data() config = tf.ConfigProto() sess = tf.Session(config=config) with gfile.FastGFile('.

如何使用C#将Tensorflow训练的.pb文件用在生产环境详解

12-31

前言 TensorFlow是Google开源的一款人工智能学习系统...这个貌似脱离不了tensorflow框架，而且生成的ckpt文件比较大，发布到生产环境的时候，还得把python的算法文件一起搞上去，如何和其他程序交互，可能还得自己去写

Tensorflow导出pb模型，并在python和matlab下分别进行预测

megaoliyuanzhende的博客

07-16

980

tensorflow下训练完模型测试程序比较杂乱，特此整理一下。 1、我都是在linux下训练，windows下测试分析，训练保存模型如下所示。 2、然后调用frozen_model.py将模型进行固化

tensorflow打印模型图_tensorflow打印pb、ckpt模型的参数以及在tensorboard里显示图结构...

weixin_39583162的博客

12-20

1664

打印pb模型参数及可视化结构import tensorflow as tffrom tensorflow.python.framework import graph_utiltf.reset_default_graph() # 重置计算图output_graph_path = '/home/huihua/NewDisk/stuff_detector_v1.pb'with tf.Session()...

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

09-18

在TensorFlow中，PB文件通常用来存储训练好的模型或模型的一部分。这种文件格式非常适合用于模型的持久化存储，因为它们体积小、读写速度快。 ##### 如何加载并打印PB模型的所有节点 1. **导入必要的库** - `...

TensorFlow把ckpt转化为pb文件代码

12-13

把tensorflow保存的checkpoint类型模型冻结，转化为.pb模型输出。调用方法参见我的博文https://blog.csdn.net/weixin_41864878/article/details/84957681

将tensorflow模型打包成PB文件及PB文件读取方式

09-18

今天小编就为大家分享一篇将tensorflow模型打包成PB文件及PB文件读取方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

Tensorflow保存和读入pb文件

Mmagic1的博客

04-30

1234

tensorflow保存pb文件 import tensorflow as tf import os from tensorflow.python.framework import graph_util #消除警告：Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 ...

Tensorflow 读取frozen graph（.pb格式）模型并运行

PianGe_zyl的博客

08-21

4016

软件版本： tensorflow == 1.12.0 python == 3.6.1 关于cpkt模型的加载与使用可以看这个关于.pb模型不同于cpkt的是.pb模型将模型参数和网络结构固化在同一个文件中，使用中无需分别读取参数和结构，通过tensorflow自带函数可一并读取至graph，之后操作graph即可完成运算。模型保存 https://blog.csdn.net/u014568072/article/details/85281769 模型读取首先将模型文件存放至代码的同级目录，我的模型文

tensorflow模型打包成PB文件及PB文件读取

zchenack个人专栏

11-02

6469

1. tensorflow模型文件打包成PB文件 import tensorflow as tf from tensorflow.python.tools import freeze_graph with tf.Graph().as_default(): with tf.device("/cpu:0"): config = tf.ConfigProto(allow_s...

tensorflow-pb导出与导入

qq_34027339的博客

03-25

1337

文章目录1. pb文件固化-python2. pb文件恢复-python版本3. pb文件导入-C++版本 1. pb文件固化-python 为了实现用python训练网络，用C++直接运行网络，因此需要将网络固化成pb文件输出。我的pb固化是在前向测试的时候实现的。代码如下 with tf.name_scope("Input"): # input_images = tf.pla...

tensorflow2读取frozen_model.pb中的节点名字

wilbur520的博客

09-19

482

import tensorflow as tf from tensorflow import keras import os model_dir = '/Users/demo/Documents/models/test3/' model_name = 'frozen_graph.pb' def create_graph(): with tf.io.gfile.GFile(os.path.join(model_dir, model_name), 'rb') as f: grap

【Tensorflow】模型存储，生成“pb”文件与读取

AI知识搬运工

09-13

4597

#通过convert_variables_to_constants函数将计算图中的变量及其取值通过常量的方式保存于一个文件中 import tensorflow as tf from tensorflow.python.framework import graph_util v1 = tf.Variable(tf.constant(1.0, shape=[1]), name="v1") ...

tensorflow pb文件如何训练

12-02

使用TensorFlow训练.pb文件的过程分为以下几个步骤： 1. 准备数据：根据你的任务需要，准备好用于训练的数据集。可以将数据集进行预处理，如划分训练集、验证集和测试集，并进行数据增强或标准化等操作。 2. 构建...