从TensorFlow模型中提取具体参数值

最新推荐文章于 2022-04-18 21:53:44 发布

NeverGx

最新推荐文章于 2022-04-18 21:53:44 发布

阅读量2.6k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/NeverGx/article/details/88919543

本文介绍了一种使用TensorFlow从checkpoint文件中读取模型参数的方法。通过获取模型检查点的状态，并利用pywrap_tensorflow模块的CheckpointReader类，可以方便地获取模型中所有变量的名称及其形状，并展示如何读取特定变量的具体数值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码
import tensorflow as tf
from tensorflow.python import pywrap_tensorflow

model_dir = “/home/model”

ckpt = tf.train.get_checkpoint_state(model_dir)
ckpt_path = ckpt.model_checkpoint_path

reader = pywrap_tensorflow.NewCheckpointReader(ckpt_path)
param_dict = reader.get_variable_to_shape_map()
print(param_dict)
print(reader.get_tensor(‘batch_normalization/gamma’).shape)

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

NeverGx

关注关注

0
点赞
踩
2

收藏

觉得还不错? 一键收藏
1
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

使用Tensorflow框架完美保存并实现调用训练好的模型（是模型不是参数哦、全网首篇）

Rainmaple‘s Blog

05-26

3万+

众所周知，Tensorflow在机器学习、深度学习领域有着广泛的应用。博主最近就Tensorflow进行了简单的学习，当我完成简单的回归模型以及CNN的训练后，一个想法油然而生，怎样将训练好的模型保存呢？保存之后又该怎么调用呢？于是，我查阅了很多相关资料，经过很多的测试，终于…成功了！现将如何保存和调用训练好的模型总结如下（提一下，本文以CNN为例了，数据集是众所周知的MNIST）： 1...

提取tensorflow中checkpoint模型参数到npy

01-03

tensorflow将训练好的网络模型保存到了checkpoint中，如果要利用训练好的网络模型去fintune另一个网络，需要为前置层的参数赋值，因此把checkpoint中的参数保存到npy文件中方便赋值。

1 条评论您还未登录，请先登录后发表或查看评论

【Pytorch】查看模型某一层的参数数值（自用）

weixin_43799388的博客

04-13

9468

参考自：pytorch获取模型某一层参数名及参数值方式 import os import torch import torch.nn as nn # 设置GPU os.environ['CUDA_VISIBLE_DEVICES'] = '1' device = torch.device('cuda:0') if torch.cuda.is_available() else 'cpu' # 创建模型 model = nn.Sequential(nn.Conv2d(3, 16, kernel_size=1)

【TensorFlow】从训练好的模型中获取参数值

热门推荐

zwwhsxq的博客

05-31

1万+

checkpoint_file = tf.train.latest_checkpoint(FLAGS.checkpoint_dir)graph = tf.Graph()with graph.as_default(): session_conf = tf.ConfigProto( allow_soft_placement=FLAGS.allow_soft_placement, ...

获得tensorflow模型的参数量

zseqsc_asd的博客

03-15

2220

tensorflow模型训练好之后，通常会保存为.ckpt文件，有时我们想了解一下模型保存了多少参数，这固然可以手动计算，但是速度太慢，这里我写了一个程序可以直接获得模型的参数量，下面是源代码：转自： https://blog.youkuaiyun.com/chnguoshiwushuang/article/details/81588794 from tensorflow.python import pyw...

TensorFlow模型保存和提取的方法

09-20

以下是对TensorFlow模型保存和提取的详细解释：首先，TensorFlow提供了`tf.train.Saver`类来实现模型的保存和恢复。在保存模型时，我们创建一个`Saver`对象，并调用其`save`方法，如`saver.save(sess,"Model/model...

tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式

09-18

TensorFlow 是一个强大的深度学习框架，它提供了获取预训练模型参数并将其应用到自定义网络结构中的功能。下面将详细介绍如何在 TensorFlow 中实现这一操作。首先，你需要导入必要的库，包括 TensorFlow 自身以及...

从训练好的tensorflow模型中打印训练变量实例

09-18

本文将详细讲解如何从训练好的TensorFlow模型中提取并打印这些训练变量。首先，我们需要导入必要的库，即TensorFlow库，用于操作和读取模型： ```python import tensorflow as tf ``` 当你完成了一次TensorFlow...

浅谈tensorflow中张量的提取值和赋值

09-18

- `tf.gather(params, indices, validate_indices=None, name=None, axis=0)`：这个函数允许你根据提供的索引`indices`从`params`张量中提取值。`axis`参数指定了沿着哪个轴进行索引。如果`indices`是标量，它将直接...

tensorflow1.X读取checkpoint(加载训练好的模型及参数)的两种方式

weixin_43220532的博客

03-25

3039

在模型训练的保存路径下面会保存有多个模型（如下图），下面介绍两种读取方式 1，获取最新的保存的一个模型，比如上图的model-5200 # model_path是checkpoints的路径，下面方法会自动寻找最新保存的一个模型 ckpt_file = tf.train.latest_checkpoint(model_path) saver.restore(sess,ckpt_file) 2，获取某个迭代次数的模型，比如要指定获取上图的model-5199 ckpt_file=os.path.join(

python3 timedelta的参数及其使用

yutu75的博客

11-10

3367

timedelta 只能计算如下参数的 timedelta参数如下: days=0 天数, seconds=0 秒数, microseconds=0 微秒;百万分之一秒, milliseconds=0 毫秒;千分之一秒, minutes=0 分钟数, hours=0 小时数, weeks=0 周数; 参数为1表示七天， from datetime import datetime, timedelta now = datetime.now() # 获取当

tensorflow2.x 索引与切片

lllfor的博客

03-14

121

举例说明索引数据中存有5张28x28的照片的三通道分析数据 data:[5, 28, 28, 3] #使用随机分布进行模拟图片数据 data = tf.random.normal([5, 28, 28, 3]) #索引至第5张照片的所有信息 data1 = data[4] 切片省略号(…) 数据中有多个维度时，只需要在其中个别维度中进行获取数据，即可用省略号代表其他维度不填写 #以上述照片的数据为例 data2 = data [3,...,2]#取第一维度中的下标为3且第四维度中下标为2的

keras.callbacks.ModelCheckpoint()

baixue0729的博客

07-17

5376

每个纪元后保存模型。 filepath可以包含命名格式化选项，它将填充日志中的纪元和键的值（在on_epoch_end中传递）。例如：如果filepath是权重。{epoch：02d} - {val_loss：.2f} .hdf5，那么模型检查点将与文件名中的纪元号和验证丢失一起保存。 def __init__(self, filepath: Any, ...

TensorFlow-断点续训

LMengi000的博客

05-13

514

import tensorflow as tf import os mnist = tf.keras.datasets.mnist (x_train,y_train),(x_test,y_test) = mnist.load_data() x_trian,x_test = x_train / 255.0,x_test / 255.0 model = tf.keras.models.Sequential([ tf.keras.layers.Flatten(), tf.keras....

07-TensorFlow 模型参数提取

鸣鼓ming的博客

04-18

1087

之前介绍了神经网络可以生成模型文件进行存储, 其实模型中的参数是可以查看的, 详情请看代码 import tensorflow as tf from tensorflow.keras.layers import Dense from tensorflow.keras import Model from sklearn import datasets import numpy as np x_train = datasets.load_iris().data y_train = datasets.load_

【tensorflow2caffe】二、tensorflow模型介绍以及参数提取

zl3090的博客

04-05

936

在博客开始前，首先郑重感谢博主jiongnima的博客对我的帮助，我按照博主的方法顺利的把bcnn模型的tensorflow版转为caffe版，关于bcnn模型，它是个细粒度分类模型，我将在后续的博客中详细介绍。首先隆重呈上博主的链接： https://blog.youkuaiyun.com/jiongnima/article/details/78326515 转模型系列博文的操作主要基于博主...

tensorflow一些参数的详细讲解分享（tensorflow1的哦）

Aaadsda414114的博客

12-19

2131

tf1: tf.GradientTape()函数： tf提供的自动求导函数 x = tf.Variable(initial_value=3.) with tf.GradientTape() as tape: # 在 tf.GradientTape() 的上下文内，所有计算步骤都会被记录以用于求导 y = tf.square(x) y_grad = tape.gradient(y, x) # 计算y关于x的导数 tf.GradientTape() 是一个自动求导的记...

tensorflow 从模型中读出参数

活到老、学到老

07-30

3117

# Create some variables. v1 = tf.get_variable("v1", shape=[3], initializer = tf.zeros_initializer) v2 = tf.get_variable("v2", shape=[5], initializer = tf.zeros_initializer) inc_v1 = v1.assign(v1+1) ...

tensorflow保存模型、加载模型和提取模型参数和特征图

walden的博客

06-24

7065

1.tf.train.latest_checkpoint('./model_data/')这一句最终返回的是一个字符串，比如'./model_data/model-99991'这个方法本身还会做相应的检查，比如checkpoint中最新的模型model_checkpoint_path中的名字在all_model_checkpoint_paths并没有，而是人为修改后的，那么就会提示：ERROR:t...