Tensorflow读取并输出已保存模型的权重数值方式
作者：AManFromEarth 发布时间：2023-09-15 10:07:49
标签：Tensorflow,输出,模型,权重数值
这篇文章是为了对网络模型的权重输出，可以用来转换成其他框架的模型。
import tensorflow as tf
from tensorflow.python import pywrap_tensorflow
#首先，使用tensorflow自带的python打包库读取模型
model_reader = pywrap_tensorflow.NewCheckpointReader(r"model.ckpt")
#然后，使reader变换成类似于dict形式的数据
var_dict = model_reader.get_variable_to_shape_map()
#最后，循环打印输出
for key in var_dict:
print("variable name: ", key)
print(model_reader.get_tensor(key))
结果（其中一个权重的示例）
........
variable name: InceptionV3/Mixed_7c/Branch_3/Conv2d_0b_1x1/weights
[[[[ 0.00013783 -0.00251428 0.02235526 ... -0.01409702 0.00340105
-0.00752808]
[ 0.01590012 -0.00258413 -0.00627338 ... -0.03600493 0.01220086
-0.01254225]
[-0.02394262 -0.00764508 -0.00895328 ... -0.01731405 0.03568469
0.00918952]
...
[-0.01865693 -0.00358359 -0.02342288 ... 0.00935838 0.00367858
-0.00976252]
[ 0.01297642 0.00223457 0.00652326 ... -0.00762609 -0.0136022
-0.01129473]
[-0.01395879 -0.00920246 0.01061887 ... 0.0236958 0.00041993
-0.01291134]]]]
......
来源：https://blog.csdn.net/AManFromEarth/article/details/81057577
0
投稿猜你喜欢
本文出自“Python为什么”系列，归档在 Github 上：https://github.com/chinesehuazhou/pytho- 昨天还和裕波沟通我没有准备ppt，想以另一种互动的方式来交流。后来想想，每个人都准备了，我不准备也不太好意思，呵呵。今天在webrebuil
- 对网站的LOGO设计做了一些归纳，希望得到批评，发现写的太长了，又不忍心删减，就分成两部分了，第一部分是有关设计基础的。第二部分是关于网站L
- 炸金花题目很简单：就是自己写一个程序，实现诈金花游戏的发牌、判断输赢。规则：一付扑克牌，去掉大小王，每个玩家发3张牌，最后比大小，看谁赢。牌
- python使用utf8编码，mysql也是utf8编码，是什么问题?后来查了一下，使用一个简单的办法即可：vsql = "ins
- 一. create优化在serializer序列化中，我们通过创建序列化器对象的方式地简化了视图函数的代码，前端传入的数据通过反序列化操作进
- 将有安全问题的SQL过程删除，比较全面.一切为了安全!删除了调用shell，注册表，COM组件的破坏权限use master&nb
- insert into testtable(recordnumber,currentdate) values (i,sysdate); pr
- 我希望大家看到该标题就能让想象到它的功能： 1、WITH TEMPL
- 一，进程的理论基础一个应用程序，归根结底是一堆代码，是静态的，而进程才是执行中的程序，在一个程序运行的时候会有多个进程并发执行。进程和线程的
- 目录前言1. 使用Lambda来修改Pandas数据框中的值2. 使用f-string来连接字符串3. 用Zip()函数对多个列表进行迭代4
- 前言异常值是指样本中的个别值，也称为离群点，其数值明显偏离其余的观测值。常用检测方法3σ原则和箱型图。其中，3σ原则只适用服从正态分布的数据
- Python的装饰器（decorator）是一个很棒的机制，也是熟练运用Python的必杀技之一。装饰器，顾名思义，就是用来装饰的，它装饰的
- 为了处理根据Web标准创作的网页和根据盛行于20世纪90年代末的旧时实践创作的网页，当代的Web浏览器实现了各种不同的引擎模式。本文说明了那
- 经常会在连接DB的时候用到，就是不知道代表什么意思。。。RS.OPEN SQL,CONN,A,BA: ADOPENFORWARDONLY(=
- 安装python中文分词库jieba法1：Anaconda Prompt下输入conda install jieba法2：Terminal下
- 今天我们来一起看看Dreamweaver MX 2004在加密FTP 传送 方面的新功能。我们一般在做
- networkx是Python的一个包，用于构建和操作复杂的图结构，提供分析图的算法。图是由顶点、边和可选的属性构成的数据结构，顶点表示数据
- 调用：var pageChange = function (index) { &n
- HTTP头中一般断点下载时才用到Range和Content-Range实体头，Range用户请求头中，指定第一个字节的位置和最后一个字节的位
