将tensorflow模型打包成PB文件及PB文件读取方式
作者：zchenack 发布时间：2022-07-02 22:47:33
标签：tensorflow,打包,PB文件
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_soft_placement=True)
with tf.Session(config=config).as_default() as sess:
model = Your_Model_Name()
model.build_graph()
sess.run(tf.initialize_all_variables())
saver = tf.train.Saver()
ckpt_path = "/your/model/path"
saver.restore(sess, ckpt_path)
graphdef = tf.get_default_graph().as_graph_def()
tf.train.write_graph(sess.graph_def,"/your/save/path/","save_name.pb",as_text=False)
frozen_graph = tf.graph_util.convert_variables_to_constants(sess,graphdef,['output/node/name'])
frozen_graph_trim = tf.graph_util.remove_training_nodes(frozen_graph)
freeze_graph.freeze_graph('/your/save/path/save_name.pb','',True, ckpt_path,'output/node/name','save/restore_all','save/Const:0','frozen_name.pb',True,"")
2. PB文件读取使用
output_graph_def = tf.GraphDef()
with open("your_name.pb","rb") as f:
output_graph_def.ParseFromString(f.read())
_ = tf.import_graph_def(output_graph_def, name="")
node_in = sess.graph.get_tensor_by_name("input_node_name")
model_out = sess.graph.get_tensor_by_name("out_node_name")
feed_dict = {node_in:in_data}
pred = sess.run(model_out, feed_dict)
来源：https://blog.csdn.net/hustchenze/article/details/83660960
0
投稿猜你喜欢
- 一、操作redisredis是一个key-value存储系统，value的类型包括string(字符串)，list(链表)，set(集合)，
目录一、介绍二、方法三、示例windows平台1、编写搭建动态库c函数2、python导入库3、结果linux平台1、编写c程序2、编译成s- 一、概述在一般的sql操作中，sql语句基本上都是固定的，如： SELECT t.empno,t.ename FROM scott
- 目录1、typing介绍2、typing的作用3、常用类型3.1 代码示例4、typing模块的其他用法4.1 类型别名4.2 NewTyp
- 下面是用Python实现Floyd算法的代码，供大家参考，具体内容如下# -*- coding: utf-8 -*- ""
- 1，在Python中将integer数转化为罗马数说明：在罗马数中（3999以内），和阿拉伯数字相似，可以把它分解为个位，十位，百位，千位，
- 第一章：连接 FTP 服务器并实现文件夹下载① 连接 FTP 服务器如果 FTP 不用用户名密码就直接可以访问，那就是用的默认用户名 Ano
- Python运行出现DeprecationWarning今天运行 py 时出现下面的问题：DeprecationWarning: Using
- 由 于数据库日志增长被设置为“无限制”，所以时间一长日志文件必然会很大，一个400G的数据库居然有600G的LOG文件，严重占用了磁盘空间。
- 有时候很多朋友访问自己的asp或php等程序页面时出现一些错误，就是页面无法显示也没有详细的错误信息，就算iis中开启了显示详细的错误也不能
- 引子Matlab中有一个函数叫做find,可以很方便地寻找数组内特定元素的下标，即：Find indices and values of n
- 本文实例讲述了Python设计模式之抽象工厂模式原理与用法。分享给大家供大家参考，具体如下：抽象工厂模式(Abstract Factory
- 使用神经网络进行样本训练，要实现随机梯度下降算法。这里我根据麦子学院彭亮老师的讲解，总结如下，（神经网络的结构在另一篇博客中已经定义）：de
- golang切片和数组拷贝基础Go 语言里面变量有两类，一类是值类型，一类是引用类型。在 Go 语言中：string、int、bool、fl
- 首先要把php_iconv.dll和inconv.dll COPY到c:\winnt\system32下，直接上代码：<?define
- 1 unittest框架unittest 是python 的单元测试框架，它主要有以下作用：提供用例组织与执行：当你的测试用例只有几条时，可
- 如下：import pandas as pddf = pd.DataFrame({'Country':['China
- TF-IDF（term frequency–inverse document frequency）是一种用于信息检索与数据挖掘的常用加权技术
- 一、在windows环境下修改pip镜像源的方法(以python3.5为例)(1):在windows文件管理器中,输入 ％APPDATA％(
- 其实之前笔者写代码的时候用到模型的保存和加载，需要用的时候就去度娘搜一下大致代码，现在有时间就来整理下整个pytorch模型的保存和加载，开
