tensorflow 模型权重导出实例


Posted in Python onJanuary 24, 2020

tensorflow在保存权重模型时多使用tf.train.Saver().save 函数进行权重保存,保存的ckpt文件无法直接打开,不利于将模型权重导入到其他框架使用(如Caffe、Keras等)。

好在tensorflow提供了相关函数 tf.train.NewCheckpointReader 可以对ckpt文件进行权重查看,因此可以通过该函数进行数据导出。

import tensorflow as tf
import h5py

cpktLogFileName = r'./checkpoint/checkpoint' #cpkt 文件路径
with open(cpktLogFileName, 'r') as f:
  #权重节点往往会保留多个epoch的数据,此处获取最后的权重数据   
  cpktFileName = f.readline().split('"')[1]   

h5FileName = r'./model/net_classification.h5'

reader = tf.train.NewCheckpointReader(cpktFileName)
f = h5py.File(h5FileName, 'w')
t_g = None
for key in sorted(reader.get_variable_to_shape_map()):
  # 权重名称需根据自己网络名称自行修改
  if key.endswith('w') or key.endswith('biases'):
    keySplits = key.split(r'/')
    keyDict = keySplits[1] + '/' + keySplits[1] + '/' + keySplits[2]
    f[keyDict] = reader.get_tensor(key)

以上这篇tensorflow 模型权重导出实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python网络编程之TCP通信实例和socketserver框架使用例子
Apr 25 Python
python3.5实现socket通讯示例(TCP)
Feb 07 Python
Python实现字典按照value进行排序的方法分析
Dec 23 Python
Python 数据库操作 SQLAlchemy的示例代码
Feb 18 Python
python制作英语翻译小工具代码实例
Sep 09 Python
Python单元测试与测试用例简析
Nov 09 Python
django实现web接口 python3模拟Post请求方式
Nov 19 Python
python3 dict ndarray 存成json,并保留原数据精度的实例
Dec 06 Python
关于pytorch处理类别不平衡的问题
Dec 31 Python
Python如何使用PIL Image制作GIF图片
May 16 Python
小结Python的反射机制
Sep 28 Python
python实现企业微信定时发送文本消息的示例代码
Nov 24 Python
在Tensorflow中查看权重的实现
Jan 24 #Python
tensorflow求导和梯度计算实例
Jan 23 #Python
Tensorflow的梯度异步更新示例
Jan 23 #Python
在Tensorflow中实现梯度下降法更新参数值
Jan 23 #Python
Tensorflow实现部分参数梯度更新操作
Jan 23 #Python
将tensorflow模型打包成PB文件及PB文件读取方式
Jan 23 #Python
使用tensorflow显示pb模型的所有网络结点方式
Jan 23 #Python
You might like
丧钟首部独立剧集《丧钟:骑士与龙》北美正式开播,场面血腥
2020/04/09 欧美动漫
PHP 遍历XP文件夹下所有文件
2008/11/27 PHP
PHP采集利器 Snoopy 试用心得
2011/07/03 PHP
防止本地用户用fsockopen DDOS攻击对策
2011/11/02 PHP
PHP面向对象法则
2012/02/23 PHP
php防止sql注入代码实例
2013/12/18 PHP
PHP实现在线阅读PDF文件的方法
2015/06/17 PHP
轻轻松松学习JavaScript
2007/02/25 Javascript
(仅IE下有效)关于checkbox 三态
2007/05/12 Javascript
De facto standard 世界上不可思议的事实标准
2010/08/29 Javascript
javascript静态页面传值的三种方法分享
2013/11/12 Javascript
使用Angular和Nodejs、socket.io搭建聊天室及多人聊天室
2015/08/21 NodeJs
基于Vue.js的表格分页组件
2016/05/22 Javascript
让html元素随浏览器的大小自适应垂直居中的实现方法
2016/10/12 Javascript
基于vue.js轮播组件vue-awesome-swiper实现轮播图
2017/03/17 Javascript
fullPage.js和CSS3实现全屏滚动效果
2017/05/05 Javascript
Angular.js前台传list数组由后台spring MVC接收数组示例代码
2017/07/31 Javascript
利用Console来Debug的10个高级技巧汇总
2018/03/26 Javascript
使用 vue 实例更好的监听事件及vue实例的方法
2019/04/22 Javascript
浅析微信小程序自定义日历组件及flex布局最后一行对齐问题
2020/10/29 Javascript
python 运算符 供重载参考
2009/06/11 Python
Python Web框架Flask中使用七牛云存储实例
2015/02/08 Python
Python实现的桶排序算法示例
2017/11/29 Python
Django实现表单验证
2018/09/08 Python
Flask 上传自定义头像的实例详解
2020/01/09 Python
Pycharm中import torch报错的快速解决方法
2020/03/05 Python
使用tensorflow进行音乐类型的分类
2020/08/14 Python
Python try except else使用详解
2021/01/12 Python
python爬取youtube视频的示例代码
2021/03/03 Python
BNKR中国官网:带你感受澳洲领先潮流时尚
2018/08/21 全球购物
优质美利奴羊毛袜,不只是徒步旅行:Darn Tough Vermont
2018/11/05 全球购物
澳大利亚珠宝商:Shiels
2019/10/06 全球购物
数控技术与应用毕业生自荐信
2013/09/24 职场文书
经济管理自荐书
2014/06/09 职场文书
交通事故委托书范本精选
2014/10/04 职场文书
聘用合同范本
2015/09/21 职场文书