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中函数的参数定义和可变参数用法实例分析
Jun 04 Python
Python+MongoDB自增键值的简单实现
Nov 04 Python
python 线程的暂停, 恢复, 退出详解及实例
Dec 06 Python
python远程连接服务器MySQL数据库
Jul 02 Python
Python 加密与解密小结
Dec 06 Python
pyqt 实现为长内容添加滑轮 scrollArea
Jun 19 Python
python粘包问题及socket套接字编程详解
Jun 29 Python
对python 中re.sub,replace(),strip()的区别详解
Jul 22 Python
Python3进制之间的转换代码实例
Aug 24 Python
python 利用已有Ner模型进行数据清洗合并代码
Dec 24 Python
python 字符串的驻留机制及优缺点
Jun 19 Python
Python基于locals返回作用域字典
Oct 17 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
JoshChen_web格式编码UTF8-无BOM的小细节分析
2013/08/16 PHP
php读取出一个文件夹及其子文件夹下所有文件的方法示例
2017/06/15 PHP
django中的ajax组件教程详解
2018/10/18 PHP
thinkPHP+LayUI 流加载实现功能
2019/09/27 PHP
MooTools 1.2中的Drag.Move来实现拖放
2009/09/15 Javascript
jQuery判断checkbox是否选中的3种方法
2014/08/12 Javascript
JavaScript检查弹出窗口是否被阻拦的方法技巧
2015/03/13 Javascript
浅谈javascript中return语句
2015/07/15 Javascript
JavaScript实现网站访问次数统计代码
2015/08/12 Javascript
jquery实现的用户注册表单提示操作效果代码分享
2015/08/28 Javascript
js过滤HTML标签完整实例
2015/11/26 Javascript
AngularJS ng-app 指令实例详解
2016/07/30 Javascript
javascript 初学教程及五子棋小程序的简单实现
2017/07/04 Javascript
详解VUE中常用的几种import(模块、文件)引入方式
2018/07/03 Javascript
js中的闭包实例展示
2018/11/01 Javascript
浅谈vue权限管理实现及流程
2020/04/23 Javascript
[05:53]敌法师的金色冠名ID"BurNIng",是传说,是荣耀
2020/07/11 DOTA
python在多玩图片上下载妹子图的实现代码
2013/08/13 Python
Django 导出项目依赖库到 requirements.txt过程解析
2019/08/23 Python
Python3.7 基于 pycryptodome 的AES加密解密、RSA加密解密、加签验签
2019/12/04 Python
Python实现投影法分割图像示例(二)
2020/01/17 Python
python中数据库like模糊查询方式
2020/03/02 Python
解决python运行启动报错问题
2020/06/01 Python
css3闪亮进度条效果实现思路及代码
2013/04/17 HTML / CSS
使用HTML5加载音频和视频的实现代码
2020/11/30 HTML / CSS
什么是makefile? 如何编写makefile?
2012/08/08 面试题
医药代表个人的求职信分享
2013/12/08 职场文书
基层党组织公开承诺书
2014/03/28 职场文书
党支部三严三实对照检查材料思想汇报
2014/09/29 职场文书
团队会宣传标语
2014/10/09 职场文书
爱心助学感谢信
2015/01/21 职场文书
手术室护士个人总结
2015/02/13 职场文书
寒假社会实践个人总结
2015/03/06 职场文书
初中教师德育工作总结2015
2015/05/12 职场文书
幼儿园2016年圣诞活动总结
2016/03/31 职场文书
Python实现提取PDF简历信息并存入Excel
2022/04/02 Python