19.7.26 tesorflow将ckpt文件保存为pb文件的两种方法

最新推荐文章于 2021-01-22 13:52:47 发布
最新推荐文章于 2021-01-22 13:52:47 发布
阅读量451 收藏
点赞数
分类专栏： TensorFlow 文章标签： tensorflow
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u011732139/article/details/97390723
版权

tensorflow中有两种可以将训练好的模型保存为.pb文件的模型。大家都知道一个训练好的tensorflow会产生四种文件：

  • checkpoint：用于保存模型的checkpoint路径以及所有或部分迭代一定次数之后的checkpoints文件
  • .meta文件：保存了所有变量、操作、集合等（Graph）
  • .index文件
  • .data-****-of-****文件：与 .index 文件合在一起组成了 ckpt 文件

常见的tensorflow保存模型为.pb格式的方法有两种，一是使用tf.train.Sever()，另外一种是fronze model。有关两种方法的使用，有大佬已经整理过了就不细说了。
传送门：《tensorflow保存模型的两种方法》

MQTXWD
关注
专栏目录
TensorFlow实现checkpoint文件转换为pb文件
12-20
由于项目需要，需要将TensorFlow保存的模型从ckpt文件转换为pb文件。 import os from tensorflow.python import pywrap_tensorflow from net2use import inception_resnet_v2_small#这里使用自己定义的模型函数即可 import tensorflow as tf if __name__=='__main__': pb_file = "./model/output.pb" ckpt_file = "./model/model.ckpt-652900" ''' 这里的节点名字可能
TensorFlow：将ckpt文件固化成pb文件
qq26983255的博客
09-26 1万+
本文是将yolo3目标检测框架训练出来的ckpt文件固化成pb文件，主要利用了GitHub上的该项目。 为什么要最终生成pb文件呢？简单来说就是直接通过tf.saver保存行程的ckpt文件其变量数据和图是分开的。我们知道TensorFlow是先画图，然后通过placeholde往图里面喂数据。这种解耦形式存在的方法对以后的迁移学习以及对程序进行微小的改动提供了极大的便利性。但是...
faster_rcnn训练权重ckpt转pb文件
qq_43100178的博客
05-23 1972
faster_rcnn训练权重ckpt转pb文件 借鉴大佬网址：https://www.cnblogs.com/zerotoinfinity/p/10242849.html
使用tfrecord喂数据到graph到模型训练到模型Saver到存储为pb格式文件
luoyexuge的专栏
04-11 3093
我看现在网络上大多数据教程或者demo用tensorflow训练一个模型，喂数据使用tf.placeholder，但是在大型的文本或者图像数据里面使用placeholder就不wok，性能和内存都是瓶颈。还有大多数代码模型保存什么都没有，更别说模型部署，本篇博客从原始数据转化为tfrecord  ，再把tfrecord数据丢给模型训练，再到模型保存，这里把保存好的模型再转化为pb格式文件模型，模型...
C++学习之路 | PTA乙级—— 1006 换个格式输出整数 (15分)（精简）
Eighteen
09-22 184
1006 换个格式输出整数 (15分) 让我们用字母 B 来表示“百”、字母 S 表示“十”，用 12…n 来表示不为零的个位数字 n（<10），换个格式来输出任一个不超过 3 位的正整数。例如 234 应该被输出为 BBSSS1234，因为它有 2 个“百”、3 个“十”、以及个位的 4。 输入格式： 每个测试输入包含 1 个测试用例，给出正整数 n（<1000）。 输出格式： 每个测试用例的输出占一行，用规定的格式输出 n。 输入样例 1： 234 输出样例 1： BBSSS1234 输入样
tensorflow模型文件(ckpt)转pb文件的方法（不知道输出节点名）
09-17
当我们要将ckpt转换为pb文件时，通常有两种方法： 1. 使用`tensorflow.python.tools.freeze_graph.freeze_graph` 这是更常见的一种方法，即使不知道输出节点名，只要原始模型代码可用，也可以完成转换。以下是一个...
tensorflow实现将ckpt转pb文件的方法
09-17
在TensorFlow中，模型的保存和加载是通过`tf.train.Saver()`类完成的，它会将模型的计算图结构和变量的值分别保存在不同的文件中，包括`.meta`、`.data`和`.index`等。然而，在某些场景下，如在Android或iOS等移动...
tensorflow从ckpt和从.pb文件读取变量的值方式
09-16
在TensorFlow中，模型训练完成后通常会保存为两种主要格式：`.ckpt`（checkpoint）和`.pb`（protobuf）。这两种格式都有各自的用途和读取方式。 1. **从`.ckpt`文件读取变量的值** `.ckpt`文件是TensorFlow的检查...
tensorflow的ckpt文件转pb模型文件
10-28
使用这个python程序可以直接将ckpt文件转成固化的模型文件，用于预测结果，注意修改路径
深度学习之格式转换笔记(二)：CKPT 转换成 PB格式文件
源的博客
01-22 1652
我们使用tf.train.saver()保存模型时会产生多个文件，也就是说把计算图的结构和图上参数取值分成了不同的文件存储。这也是在tensorflow中常用的保存方式。 保存文件的代码: import tensorflow as tf # 声明两个变量 v1 = tf.Variable(tf.random_normal([1, 2]), name="v1") v2 = tf.Variable(tf.random_normal([2, 3]), name="v2") init_op = tf.global_
TensorFlow把ckpt转化为pb文件代码进阶版
12-13
把tensorflow保存的checkpoint类型模型冻结，转化为.pb模型输出。调用方法参见我的博文https://blog.csdn.net/weixin_41864878/article/details/84957681
将tensorflow模型文件：*.data-00000-of-00001，*.index 和 *.meta这3个文件固化为pb文件
07-10
将tensorflow的PNet,RNet,ONet的模型文件：*.data-00000-of-00001，*.index 和 *.meta这3个文件固化为pb文件。里面详细介绍了节点名称：output_node_names如何获取。直接拿来就可以运行。可以直接看看自己模型文件的所有节点名称。代码里有详细说明
tensorflow实现将ckpt转pb文件
热门推荐
AI吃大瓜的博客
09-01 8万+
tensorflow实现将ckpt转pb文件 【尊重原创，转载请注明出处】：https://blog.csdn.net/guyuealian/article/details/82218092 本博客实现将自己训练保存的ckpt模型转换为pb文件，该方法适用于任何ckpt模型，当然你需要确定ckpt模型输入/输出的节点名称。 目录 tensorflow实现将ckpt转pb文件 一、C...
Tensorflow-pb保存与导入
风吴痕的博客
09-20 2万+
参考： 如何用Tensorflow训练模型成pb文件和和如何加载已经训练好的模型文件 tensorflow存储图和训练的权重为.pb，然后读取.pb并使用 TensorFlow模型保存和提取方法 0、基础 # 保存图表并保存变量参数 from tensorflow.python.framework import graph_util var_list=tf.global...
TensorFlow ckpt模型文件转换为pb模型文件
罗百万当程序员
06-05 5096
问题 ckpt转pb主要有两点要注意 1.要知道模型输入输出的节点名称 2.转化后Value Error问题 1.获取输入输出节点名称 使用tools里的freeze_graph来读取ckpt 保存ckpt模型的文件夹下的三个文件名 epoch_50.ckpt.data-00000-of-00001 epoch_50.ckpt.index epoch_50.ckpt.meta 获取所有节点名称 ...
训练模型转pb文件
lgq1558101442的博客
08-16 690
nasnet-tensorflow 点击连接
tensorflow--使用freeze_graph.py将ckpt转为pb文件
大丈夫
10-25 4万+
这节是关于tensorflow的Freezing，字面意思是冷冻，可理解为整合合并；整合什么呢，就是将模型文件和权重文件整合合并为一个文件，主要用途是便于发布。 官方解释可参考：https://www.tensorflow.org/extend/tool_developers/#freezing 这里我按我的理解翻译下，不对的地方请指正： 有一点令我们为比较困惑的是，tensorflow在训练
将ckpt文件生成pb文件的详细过程,并调用pb文件进行模型预测.
大白爱吃菠菜
06-26 5451
前段时间搭建了一个分类网络模型,然后用自己的数据进行了800epoch’的训练,最后默认生成了三个ckpt文件.由于要同时运行几个网络,所以打算将这个网络模型进行固化成pb文件,然后直接调用. 主要包括一下内容: 1.查看ckpt模型的输入输出张量名称. 2.将ckpt文件生成pb文件. 3.查看生成的pb文件的输入输出节点 4.运行pb文件,进行网络预测 1.查看ckpt网络的输入输出张量名...
TensorFlow：详解如何将ckpt模型转换为pb文件
"本文主要探讨如何使用TensorFlow将ckpt模型转换为.pb文件，这是一个常见的需求，特别是当需要在不同环境中部署模型或者简化模型文件管理时。.pb文件是一种包含了模型结构和权重数据的二进制格式，易于在不支持...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值