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
投稿猜你喜欢
- 导言（Introduction）这个提案描述了如何在jQuery的核心库中增加模板支持。更为特别是，这个提案描述了一个新的jQuery方法－
如果你经常与Excel或Word打交道，那么从两份表格/文档中找到不一样的元素是一件让人很头疼的工作，当然网上有很多方法、第三方软件教你如何- 安装Pytorch在pytorch官网上选择相应选项，我的是OS X, pip, python2.7, none CUDA。（之所以用pyt
- 本文实例讲述了Python生成器定义与简单用法。分享给大家供大家参考，具体如下：一、什么是生成器在Python中，由于受到内存的限制，列表容
- Oracle过程和函数相信大家都比较了解，下面就为您详细介绍Oracle过程和函数二者之间的区别，希望可以让您对Oracle过程和函数有更深
- 在一个项目中,制作呃echart图表的时候，遇到一个需求，需要从后端接口获取数据----售票员的姓名和业绩所以需要在订单表中，获取不同售票员
- 1 动机greenlet 包是 Stackless 的副产品，其将微线程称为 “tasklet” 。tasklet运
- 开发测试的时候，经常需要填写姓名、手机号、身份证号、银行卡号，既要符合格式要求、又不能重复。大家会到网上搜各种生成器。下面小编自己写的一个j
- 相对于numpy、TensorFlow、pandas这些已经经过多年维护、迭代，对于大多数Python开发者耳熟能详的库不同。今天要给大家介
- 今天尝试着将引用文献的格式按照IEEE的标准重新排版，感觉手动一条一条改太麻烦，而且很容易出错，所以尝试着用Python写了一个小程序用于根
- 本文实例为大家分享了python实现年会抽奖程序的具体代码，供大家参考，具体内容如下发一下自己写的公司抽奖程序。需求：公司年会要一个抽奖程序
- 这几天花了点时间,将把django开发好的web项目部署到Apache上，参考了官方的一些文档和互联网上的文档，还是花了比较多的时间，这里把
- 一、前言春节即将来临，大家肯定各种掏腰包花花花，小编相信大家在支付时候，微信、支付宝支付肯定是优先选择。今天小编心血来潮，为大家带来一个很有
- Python当中并无switch语句，本文研究的主要是通过字典实现switch语句的功能，具体如下。switch语句用于编写多分支结构的程序
- 本文实例讲述了php使用pthreads v3多线程实现抓取新浪新闻信息。分享给大家供大家参考，具体如下：我们使用pthreads，来写一个
- 一、概念我们可以将工作池理解为线程池。线程池的创建和销毁非常消耗资源，所以专门写一个pool，每次用过的线程池再放回pool中而不是销毁。不
- 运行效果：完整代码from tkinter import *import tkinter.messageboximport base64fr
- 前言优化随机森林算法，正确率提高1％~5％（已经有90％+的正确率，再调高会导致过拟合）论文当然是参考的，毕竟出现早的算法都被人研究烂了，什
- 近段时间由于修改一个ASP程序(有SQL注入漏洞)，在网上找了很多相关的一些防范办法，都不近人意，所以我将现在网上的一些方法综合改良了一下，
- 本文实例讲述了python中requests模块的使用方法。分享给大家供大家参考。具体分析如下：在HTTP相关处理中使用python是不必要
