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 09 Python
wxpython学习笔记(推荐查看)
Jun 09 Python
python实现文本文件合并
Dec 29 Python
Python selenium文件上传方法汇总
Nov 19 Python
python画柱状图--不同颜色并显示数值的方法
Dec 13 Python
python读取文件名并改名字的实例
Jan 07 Python
tornado+celery的简单使用详解
Dec 21 Python
Python2 与Python3的版本区别实例分析
Mar 30 Python
Jupyter notebook如何修改平台字体
May 13 Python
python3.7添加dlib模块的方法
Jul 01 Python
基于django2.2连oracle11g解决版本冲突的问题
Jul 02 Python
python装饰器代码解析
Mar 23 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 zip扩展Linux下安装过程分享
2014/05/05 PHP
php用户注册时常用的检验函数实例总结
2014/12/22 PHP
php实现的AES加密类定义与用法示例
2018/01/29 PHP
prototype class详解
2006/09/07 Javascript
收集的网上用的ajax之chat.js文件
2007/04/08 Javascript
javascript下判断一个对象是否具有指定名称的属性的的代码
2010/01/11 Javascript
基于JQuery.timer插件实现一个计时器
2010/04/25 Javascript
js简单工厂模式用法实例
2015/06/30 Javascript
浅谈JavaScript的Polymer框架中的事件绑定
2015/07/29 Javascript
理解javascript函数式编程中的闭包(closure)
2016/03/08 Javascript
javascript html5移动端轻松实现文件上传
2020/03/27 Javascript
AngularJS 面试题集锦
2016/09/06 Javascript
基于Angular.js实现的触摸滑动动画实例代码
2017/02/19 Javascript
JS实现队列的先进先出功能示例
2017/05/10 Javascript
jQuery实现的页面遮罩层功能示例【测试可用】
2017/10/14 jQuery
详解如何使用 vue-cli 开发多页应用
2017/12/16 Javascript
react以create-react-app为基础创建项目
2018/03/14 Javascript
详解vue表单——小白速看
2018/04/08 Javascript
vue 之 .sync 修饰符示例详解
2018/04/21 Javascript
vue项目如何刷新当前页面的方法
2018/05/18 Javascript
使用Three.js实现太阳系八大行星的自转公转示例代码
2019/04/09 Javascript
微信小程序第三方框架对比 之 wepy / mpvue / taro
2019/04/10 Javascript
[35:29]Secret vs VG 2018国际邀请赛淘汰赛BO3 第三场 8.23
2018/08/24 DOTA
TensorFlow实现Softmax回归模型
2018/03/09 Python
Python 类方法和实例方法(@classmethod),静态方法(@staticmethod)原理与用法分析
2019/09/20 Python
python对象转字典的两种实现方式示例
2019/11/07 Python
python pptx复制指定页的ppt教程
2020/02/14 Python
Pyside2中嵌入Matplotlib的绘图的实现
2021/02/22 Python
Autopep8的使用(python自动编排工具)
2021/03/02 Python
IE下实现类似CSS3 text-shadow文字阴影的几种方法
2011/05/11 HTML / CSS
size?荷兰官方网站:英国高级运动鞋精品店
2020/07/24 全球购物
群众路线教育实践活动剖析材料
2014/09/30 职场文书
长城导游词
2015/01/30 职场文书
新郎结婚保证书
2015/02/26 职场文书
清洁工个人工作总结
2015/03/05 职场文书
redis不能访问本机真实ip地址的解决方案
2021/07/07 Redis