Tensorflow读取并输出已保存模型的权重数值方式


Posted in Python onJanuary 04, 2020

这篇文章是为了对网络模型的权重输出,可以用来转换成其他框架的模型。

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]]]]
......

以上这篇Tensorflow读取并输出已保存模型的权重数值方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python的内存泄漏及gc模块的使用分析
Jul 16 Python
selenium+python实现自动化登录的方法
Sep 04 Python
python的依赖管理的实现
May 14 Python
Python实现个人微信号自动监控告警的示例
Jul 03 Python
详解如何减少python内存的消耗
Aug 09 Python
Python代码生成视频的缩略图的实例讲解
Dec 22 Python
Django用户身份验证完成示例代码
Apr 03 Python
python 还原梯度下降算法实现一维线性回归
Oct 22 Python
详解matplotlib绘图样式(style)初探
Feb 03 Python
Python编解码问题及文本文件处理方法详解
Jun 20 Python
详解Python flask的前后端交互
Mar 31 Python
python中mongodb包操作数据库
Apr 19 Python
tensorflow实现打印ckpt模型保存下的变量名称及变量值
Jan 04 #Python
tensorflow 获取所有variable或tensor的name示例
Jan 04 #Python
tensorflow没有output结点,存储成pb文件的例子
Jan 04 #Python
TensorFlow查看输入节点和输出节点名称方式
Jan 04 #Python
根据tensor的名字获取变量的值方式
Jan 04 #Python
将tensorflow.Variable中的某些元素取出组成一个新的矩阵示例
Jan 04 #Python
tensorflow实现tensor中满足某一条件的数值取出组成新的tensor
Jan 04 #Python
You might like
php URL编码解码函数代码
2009/03/10 PHP
PHP中用hash实现的数组
2011/07/17 PHP
php遍历数组的方法分享
2012/03/22 PHP
ThinkPHP3.1新特性之动态设置自动完成和自动验证示例
2014/06/19 PHP
Laravel 5框架学习之表单验证
2015/04/08 PHP
wordpress自定义标签云与随机获取标签的方法详解
2019/03/22 PHP
php判断/计算闰年的方法小结【三种方法】
2019/07/06 PHP
关于删除时的提示处理(确定删除吗)
2013/11/03 Javascript
juery框架写的弹窗效果适合新手
2013/11/27 Javascript
解析JavaScript中的不可见数据类型
2013/12/02 Javascript
node.js中的fs.closeSync方法使用说明
2014/12/17 Javascript
Nginx上传文件全部缓存解决方案
2015/08/17 Javascript
探讨JavaScript标签位置的存放与功能有无关系
2016/01/15 Javascript
深入浅析JavaScript面向对象和原型函数
2016/02/06 Javascript
高效Web开发的10个jQuery代码片段
2016/07/22 Javascript
Django1.7+JQuery+Ajax验证用户注册集成小例子
2017/04/08 jQuery
bootstrap如何让dropdown menu按钮式下拉框长度一致
2017/04/10 Javascript
Dropify.js图片宽高自适应的方法
2017/11/27 Javascript
微信小程序定位当前城市的方法
2018/07/19 Javascript
JS回调函数简单易懂的入门实例分析
2019/09/29 Javascript
vue element-ui实现input输入框金额数字添加千分位
2019/12/29 Javascript
Python全局变量用法实例分析
2016/07/19 Python
利用Python批量提取Win10锁屏壁纸实战教程
2018/03/27 Python
Python实现的FTP通信客户端与服务器端功能示例
2018/03/28 Python
python 对多个csv文件分别进行处理的方法
2019/01/07 Python
Django框架视图函数设计示例
2019/07/29 Python
Laravel框架表单验证格式化输出的方法
2019/09/25 Python
HTML5边玩边学(2)基础绘图实现方法
2010/09/21 HTML / CSS
菲律宾优惠券网站:MetroDeal
2019/04/12 全球购物
求职信格式范本
2013/11/15 职场文书
财务主管岗位职责
2014/02/28 职场文书
房屋买卖委托公证书
2014/04/08 职场文书
经济贸易系求职信
2014/08/04 职场文书
优秀乡村医生先进事迹材料
2014/08/23 职场文书
利用Sharding-Jdbc进行分库分表的操作代码
2022/01/22 Java/Android
python_tkinter事件类型详情
2022/03/20 Python