[tensorflow]从Keras加载、保存模型、显示模型权重

一骑红尘荔枝来

于 2023-03-08 20:22:43 发布

阅读量994

点赞数

分类专栏： Tensorflow学习文章标签： tensorflow keras 深度学习

本文链接：https://blog.csdn.net/condom10010/article/details/129410641

版权

Tensorflow学习专栏收录该内容

10 篇文章 0 订阅

订阅专栏

该文展示了如何在TensorFlow中使用InceptionV3模型，包括加载imagenet预训练权重，将模型保存到本地./module/InceptionV3目录，以及通过两种方式加载本地保存的模型。最后，模型的结构和参数通过`base_model.summary()`进行了展示。

摘要由CSDN通过智能技术生成

import tensorflow as tf

构建模型，并加载imagenet预训练权重

base_model = tf.keras.applications.InceptionV3(include_top=False,weights='imagenet')

保存模型到本地./module/InceptionV3下

tf.saved_model.save(base_model,'./module/InceptionV3')

加载本地模型-方式1

base_model = tf.keras.applications.InceptionV3(include_top=False,weights='./module/InceptionV3')

加载本地模型-方式2

base_model = tf.keras.applications.InceptionV3(include_top=False)
base_model.load_weights('./module/InceptionV3')

显示模型的结构和参数

base_model.summary()

在这里插入图片描述

关注博主即可阅读全文

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

一骑红尘荔枝来

关注关注

0
点赞
踩
3

收藏

觉得还不错? 一键收藏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

tensorflow及其keras如何保存模型

weixin_43178406的博客

05-28

1万+

本文主要介绍tensorflow及其keras保存并加载模型的实战代码，在经过了亲身的实践后，终于找到了可复现的实战代码，最终将详细的代码总结如下，希望能对学习tensorflow2/keras的同学有所帮助。文章目录 1. keras模型保存概述 2. TensorFlow SavedModel 格式 2.1 使用高级API：tf.keras.models.save_model 2.2 使用低级API：tf.saved_model.save 3. keras HDF5格式

Tensorflow 2.x(keras)源码详解之第十章：keras中的模型保存与加载(详解Checkpoint&md5&模型序列化)

热门推荐

weixin_43178406的博客

05-18

4万+

1. 模型保存与加载：综述 2. 前提总结 3. 仅保存模型权重 3.1 在内存中将权重进行迁移 3.2 将权重保存到磁盘上 3.2.1 Checkpoint(检查点) 3.2.2 使用tf.train.checkpoint()保存多个检查点 3.2.3 使用tf.train.checkpoint加载多个检查点 3.2.4 tf.train.checkpoint保存和加载单一检查点 3.2.5 save_weights保存检查点 3.2.6 save_weights保存成HDF5格式 4. 仅保存模型架构

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

Tensorflow读取并输出已保存模型的权重数值方式

09-18

今天小编就为大家分享一篇Tensorflow读取并输出已保存模型的权重数值方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看不看

tensorflow 获取已保存模型的权重并且打印

qq_45836261的博客

01-27

2365

获取权重方法：

keras获取model权重

一定要站在自己热爱的生活里闪闪发光

06-02

2万+

方法一：通过get_layer()函数先获取要获取权重对应的层；接着通过get_weights() 函数获取改成对应的全部参数（是一个长的为2的列表，列表中的每一个元素是一个numpy数组，第一存放该层的权重W的参数，第一个存放偏置b） layer的两个函数： get_weights() set_weights(weights) model = load_model('vgg.h5') layer1 = model.get_layer(index=2) weights = layer1.get_we

keras加载模型

weixin_39965184的博客

02-09

1810

说明：该程序是一个包含两个隐藏层的神经网络。演示如何加载一个保存好的模型。数据集：MNIST 1.加载keras模块 from __future__ import print_function #Python提供了__future__模块，把下一个新版本的特性导入到当前版本，于是我们就可以在当前版本中测试一些新版本的特性。 import numpy as np np.random.seed(13...

keras中的模型加载

123

03-12

791

from keras import backend as K from keras.models import load_model from PIL import Image import numpy as np source='/storage/DuanFX/python_code/hope/' def get_layer_output(model, x, index=-1): layer = K.function([model.input], [model.layers[inde

Deep Learning Toolbox Importer for TensorFlow-Keras Models.rar

03-26

然后，通过MATLAB命令行或者图形用户界面，可以加载保存的TensorFlow或Keras模型文件。导入模型后，MATLAB会生成对应的MATLAB模型对象，用户可以利用MATLAB的函数进行模型的操作和预测。该支持包包含的...

keras 如何保存最佳的训练模型

09-16

Keras，作为TensorFlow的一个高级API，提供了方便的工具来实现这一目标。本文将详细讲解如何在Keras中保存最佳训练模型，并介绍相关参数和技巧。 1. 只保存最佳的训练模型在Keras中，你可以使用`ModelCheckpoint`...

tensorflow2.0 下keras模型的保存与加载

woshinierye的博客

06-02

1647

在前面的博文tensorflow2.0自制神经网络数据集及预测结果（混淆矩阵）可视化中，我们用了谷歌训练好的网络模型DesNet121来训练我们的图片，当用自己的电脑和自写的网络架构来训练图片时，我们需要及时的保存我们训练好的网络，以便以后测试或者接着训练。现在网上有很多的博客介绍了在keras框架下如何保存和加载训练好的模型，在此我做一个总结吧，也方便我以后查看，不用到处去别的地方找了。我将DesNet网络换为自己的resnet18以后开始训练，训练结束后在代码的最后加了一行 model.save('m

Keras 加载已经训练好的模型进行预测操作

09-16

主要介绍了Keras 加载已经训练好的模型进行预测操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

tensorflow查看权重参数值

andeyeluguo的博客

07-24

1万+

每个框架都有查看权重参数的机制，在tenstensorflow中查看的例子如下 import tensorflow as tf import numpy as np reader = tf.train.NewCheckpointReader('model-100') all_variables = reader.get_variable_to_shape_map() w0 = reader...

tensorflow和pytorch设置和查看权重

qq_45759229的博客

04-13

2210

pytorch import numpy as np import torch.nn as nn import random import os import torch import seaborn as sns import matplotlib.pyplot as plt def seed_torch(seed=42): random.seed(seed) os.environ['PYTHONHASHSEED'] = '0' np.random.seed(seed)

keras 11_加载模型

guoqingru0311的博客

10-12

179

keras 11_加载模型 import numpy as np from keras.datasets import mnist from keras.utils import np_utils from keras.models import Sequential, load_model from keras.layers import Dense from keras.optimizers import SGD # 载入数据 (x_train, y_train), (x_test, y_test)

关于tf.keras的模型加载

qq_41919659的博客

12-29

448

tf.keras模型加载

迁移学习之--tensorflow选择性加载权重

Double_V的博客

09-28

1万+

迁移学习的实现需要网络在其他数据集上做预训练，完成参数调优工作，然后拿预训练好的参数在新的任务上做fine-tune，但是有时候可能只需要预训练的网络的一部分权重，本文主要提供一个方法如何在tf上加载想要加载的权重。在使用tensorflow加载网络权重的时候，直接使用tf.train.Saver().restore(sess, ‘ckpt’)的话是直接加载了全部权重，我们可能只需要加载网络的...

载入keras训练的模型

weixin_35756373的博客

02-12

154

可以使用 Keras 提供的 load_model 函数来载入训练过的模型。 from keras.models import load_model model = load_model('model.h5') 其中 model.h5 是模型保存的文件名。在训练模型时，可以使用 model.save('model.h5') 来保存模型。 ...

机器学习19：Keras保存和加载网络模型及权重的方法总结

科学边界

04-20

5460

0.先实现手写体识别的实例：导入训练数据 from keras.datasets import mnist from keras.preprocessing.image import ImageDataGenerator from matplotlib import pyplot from keras import backend as K #K.set_image_dim_ordering(...

Keras基础自学十一（加载权重）

gamblerofdestinyR14的博客

01-30

449

one-hot encoding Y_labels = to_categorical(Y, num_classes=3) # 设定随机种子 seed = 7 np.random.seed(seed) # 构建模型函数 def load_model(optimizer='rmsprop', init='glorot_uniform'): # 构建模型 model = Sequen...

tensorflow.keras 保存模型代码？