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实现设置windows桌面壁纸代码分享
Mar 28 Python
Python中的random()方法的使用介绍
May 15 Python
python基于右递归解决八皇后问题的方法
May 25 Python
Python for Informatics 第11章 正则表达式(一)
Apr 21 Python
Python中模块pymysql查询结果后如何获取字段列表
Jun 05 Python
Python异常对代码运行性能的影响实例解析
Feb 08 Python
Python实现的远程登录windows系统功能示例
Jun 21 Python
python中bytes和str类型的区别
Oct 21 Python
python 画函数曲线示例
Dec 04 Python
dpn网络的pytorch实现方式
Jan 14 Python
python文件路径操作方法总结
Dec 21 Python
Python超简单容易上手的画图工具库推荐
May 10 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小技巧之JS和CSS优化工具Minify的使用方法
2014/05/19 PHP
PHP开发注意事项总结
2015/02/04 PHP
Laravel 连接(Join)示例
2019/10/16 PHP
PhpSpreadsheet设置单元格常用操作汇总
2020/11/13 PHP
根据分辨率不同,调用不同的css文件
2006/07/07 Javascript
Javascript的匿名函数小结
2009/12/31 Javascript
JS中 用户登录系统的解决办法
2013/04/15 Javascript
JQuery结合CSS操作打印样式的方法
2013/12/24 Javascript
CSS3,HTML5和jQuery搜索框集锦
2014/12/02 Javascript
JS实现的网页倒计时数字时钟效果
2015/03/02 Javascript
简单的分页代码js实现
2016/05/17 Javascript
assert()函数用法总结(推荐)
2017/01/25 Javascript
ajax前台后台跨域请求处理方式
2018/02/08 Javascript
vue 界面刷新数据被清除 localStorage的使用详解
2018/09/16 Javascript
vue+web端仿微信网页版聊天室功能
2019/04/30 Javascript
javascript触发模拟鼠标点击事件
2019/06/26 Javascript
vue v-for 使用问题整理小结
2019/08/04 Javascript
js canvas实现星空连线背景特效
2019/11/01 Javascript
微信小程序 SOTER 生物认证DEMO 指纹识别功能
2019/12/13 Javascript
vue随机验证码组件的封装实现
2020/02/19 Javascript
谈谈JavaScript令人迷惑的==与+
2020/08/31 Javascript
NodeJS开发人员常见五个错误理解
2020/10/14 NodeJs
vue 动态添加的路由页面刷新时失效的原因及解决方案
2021/02/26 Vue.js
在Docker上部署Python的Flask框架的教程
2015/04/08 Python
Python3使用pandas模块读写excel操作示例
2018/07/03 Python
Python入门Anaconda和Pycharm的安装和配置详解
2019/07/16 Python
使用Pycharm分段执行代码
2020/04/15 Python
html5中localStorage本地存储的简单使用
2017/06/16 HTML / CSS
html5触摸事件判断滑动方向的实现
2018/06/05 HTML / CSS
屈臣氏越南官网:Watsons越南
2021/01/14 全球购物
采购部主管岗位职责
2014/01/01 职场文书
十八届三中全会感言
2014/03/10 职场文书
管理提升方案
2014/06/04 职场文书
健康状况证明书
2014/11/26 职场文书
2016元旦主持人经典开场白台词
2015/12/03 职场文书
JavaScript 事件捕获冒泡与捕获详情
2021/11/11 Javascript