tensorflow 模型权重导出实例
更新时间：2020年01月24日 09:34:01 作者：张中星
今天小编就为大家分享一篇tensorflow 模型权重导出实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
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 模型权重导出实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
Pytorch的torch.utils.data中Dataset以及DataLoader示例详解
torch.utils.data 是 PyTorch 提供的一个模块，用于处理和加载数据，该模块提供了一系列工具类和函数，用于创建、操作和批量加载数据集，这篇文章主要介绍了Pytorch的torch.utils.data中Dataset以及DataLoader等详解,需要的朋友可以参考下2023-08-08
这篇文章主要介绍了python中的变量命名规则详情，变量名可以包括字母、数字、下划线，但是数字不能做为开头，变量用的好或不好，和代码质量有着非常重要的联系，合理的使用变量，可以让你的代码可读性更高并且更加简洁,下面相关内容吧需要的小伙伴可以参考一下2022-03-03
这篇文章主要给大家介绍了关于Odoo中如何生成唯一不重复的序列号的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧。2018-02-02
这篇文章主要介绍了Python tuple方法和string常量，文章基于python的相关资料展开详细内容，对初学python的通知有一定的参考价值，需要的小伙伴可以参考一下2022-05-05
在本篇文章里小编给大家整理的是关于Python各种扩展名区别点整理，需要的朋友们可以学习下。2020-02-02
这篇文章主要介绍了python的unittest测试类代码实例，具有一定参考价值，需要的朋友可以了解下。2017-12-12
这篇文章主要介绍了python 三种方法实现对Excel表格的读写，帮助大家更好的利用python处理表格，感兴趣的朋友可以了解下2020-11-11
这篇文章主要介绍了用Python爬虫+数据分析+数据可视化，分析《雪中悍刀行》弹幕,本文很适合初学python的同学入门阅读，需要的朋友可以参考下2022-01-01
python用sqlacodegen根据已有数据库(表)结构生成对应SQLAlchemy模型
本文介绍了如何使用sqlacodegen获取数据库所有表的模型类，然后使用ORM技术进行CRUD操作，有此需求的朋友可以了解下本文2021-06-06
这篇文章主要介绍了Python通过类的组合模拟街道红绿灯,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-09-09
最新评论