tensorflow 模型权重导出实例
作者：张中星 发布时间：2022-10-14 06:25:25
标签：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）
来源：https://www.cnblogs.com/zhangzhongxing/p/9258677.html
0
投稿猜你喜欢
- html 页面<html lang="en"><head> <meta charset=&
python安装教程和Pycharm安装详细教程，分享给大家。首先我们来安装python1、首先进入网站下载：点击打开链接（或自己输入网址h- 人的大脑通过双眼来辨别视觉图形获取信息。大脑根据储存的经验，将所看到的视觉图形建立起优先级。由此可见，一个良好的视觉设计可以帮助大脑迅速有效
- 使用python的numpy模块实现逻辑回归模型的代码，供大家参考，具体内容如下使用了numpy模块，pandas模块，matplotlib
- 1. views.py定义views视图函数，将数据存入字典。并用压缩为json格式，dumps，并return。import jsonde
- 代码如下所示：scole = input("input your scole:")if scole>90: &nb
- 双向链表的基本操作的实现，供大家参考，具体内容如下在之前的博客中介绍了三种链表，分别是单链表、单向循环链表以及双向链表。本篇博客将用Pyth
- 本文实例讲述了php打包压缩文件之ZipArchive方法用法。分享给大家供大家参考，具体如下：前面说到了php打包压缩文件之PclZip方
- 一、 什么是进程 / 线程1、 引论众所周知，CPU是计算机的核心，它承担了所有的计算任务。而操作系统是计算机的管理者，是一个大管家，它负责
- 类的参数定义将conda环境设置为ai，conda activate ai这个文件的由来：由于在yolov1的pytorch实现的损失函数中
- MySQL两张表取差集业务场景如下：人员表中有证件号、手机号字段，这两个字段因为涉及到个人隐私问题，因此加密存储，有另外一张解密表可以和人员
- 最近在使用Tensorflow 实现DNN网络时，遇到一些问题。目前网上关于Tensorflow的资料还比较少，现把问题和解决方法写出来，仅
- 前言Vscode是是一个强大的跨平台工具，我自己电脑是mac，公司电脑是win而且是内部环境，导致公司安装软件很费劲。好在vscode许多插
- 实现代码# batch_handle_image.pyimport argparseimport globimport osfrom PIL
- 刚开始使用webpack时，可能很多人都会有过这样的想法，在require文件时，能不能不写静态的字符串路径，而是使用一个更灵活的方式，比如
- 昨天在得写“RSS收藏”栏目展现的时候我让RSS的列表用弹出的形式展现了，开始是直接的控制 style.display 来做的，但是感觉有点
- “你如何为成千上万的用户和页面提供CSS？” 这是Nicole Sullivan在她的在丹佛的Web Directions North 大会
- 注意：使用该方法，获取的数据总数目不能超过1万，否则出错#在python3上运行from elasticsearch import Elas
- 一、Time 包中定时器函数go v1.20.4定时函数：NewTicker，NewTimer 和 time.After 介绍time 包中
- 在分析查询性能时，考虑EXPLAIN关键字同样很管用。EXPLAIN关键字一般放在SELECT查询语句的前面，用于描述MySQL如何执行查询
