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的一些用法分享
Oct 07 Python
python实现随机森林random forest的原理及方法
Dec 21 Python
django manage.py扩展自定义命令方法
May 27 Python
详解通过API管理或定制开发ECS实例
Sep 30 Python
Python数据类型之List列表实例详解
May 08 Python
pyqt5 lineEdit设置密码隐藏,删除lineEdit已输入的内容等属性方法
Jun 24 Python
使用Python项目生成所有依赖包的清单方式
Jul 13 Python
python自动化测试三部曲之request+django实现接口测试
Oct 07 Python
Python descriptor(描述符)的实现
Nov 15 Python
解决python 在for循环并且pop数组的时候会跳过某些元素的问题
Dec 11 Python
python 爬取吉首大学网站成绩单
Jun 02 Python
Python内置的数据类型及使用方法
Apr 13 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
桌面中心(一)创建数据库
2006/10/09 PHP
php设置允许大文件上传示例代码
2014/03/10 PHP
PHP 数据结构队列(SplQueue)和优先队列(SplPriorityQueue)简单使用实例
2015/05/12 PHP
Yii2中事务的使用实例代码详解
2016/09/07 PHP
javascript和jquery实现设置和移除文本框默认值效果代码
2015/01/13 Javascript
coffeescript使用的方式汇总
2015/08/05 Javascript
JavaScript多并发问题如何处理
2015/10/28 Javascript
轻松使用jQuery双向select控件Bootstrap Dual Listbox
2015/12/13 Javascript
Atitit.js的键盘按键事件捆绑and事件调度
2016/04/01 Javascript
Bootstrap每天必学之折叠
2016/04/12 Javascript
详解Angular开发中的登陆与身份验证
2016/07/27 Javascript
Nodejs实现爬虫抓取数据实例解析
2018/07/05 NodeJs
react-native动态切换tab组件的方法
2018/07/07 Javascript
微信小程序自定义音乐进度条的实例代码
2018/08/28 Javascript
JavaScript实现简单的隐藏式侧边栏功能示例
2018/08/31 Javascript
js中实例与对象的区别讲解
2019/01/21 Javascript
vue vant Area组件使用详解
2019/12/09 Javascript
python 切片和range()用法说明
2013/03/24 Python
Python编程之string相关操作实例详解
2017/07/22 Python
Python栈算法的实现与简单应用示例
2017/11/01 Python
python随机数分布random测试
2018/08/27 Python
Django实现列表页商品数据返回教程
2020/04/03 Python
keras中epoch,batch,loss,val_loss用法说明
2020/07/02 Python
python 实现有道翻译功能
2021/02/26 Python
Bose法国官网:购买耳机、扬声器、家庭影院、专业音响
2017/12/21 全球购物
英国DIY和家居装饰领域的主要品牌:Wickes
2019/11/26 全球购物
北京某公司的.net笔试题
2014/03/20 面试题
函授毕业生的自我鉴定
2013/11/26 职场文书
法律进机关实施方案
2014/03/12 职场文书
青年安全生产示范岗事迹材料
2014/05/04 职场文书
行政管理专业求职信
2014/07/06 职场文书
党的群众路线教育实践活动学习笔记
2014/11/05 职场文书
护理专业自我评价
2015/03/11 职场文书
党小组鉴定意见
2015/06/02 职场文书
css中z-index: 0和z-index: auto的区别
2021/08/23 HTML / CSS
SQL Server查询某个字段在哪些表中存在
2022/03/03 SQL Server