keras 网络结构和权值文件的保存与载入

小咪咪 2022-05-27 01:07 156阅读 0赞

以lenet网络为例

from keras.models import Model
    from keras.layers import Input,Conv2D,Dense,Flatten,MaxPooling2D,Activation
    ####################### net structure ##########################
    inp=Input(shape=(28,28,1))
    out=Conv2D(20,(5,5))(inp)
    out=MaxPooling2D(pool_size(2,2))(out)
    out=Conv2D(50,(5,5))(inp)
    out=MaxPooling2D(pool_size(2,2))(out)
    out=Flatten()(out)
    out=Dense(500)(out)
    out=Activation('relu')(out)
    out=Dense(2)(out)
    out=Activation('softmax')(out)
    #################################################################
    model=Model(inp,out)
    model.compile(
    			optimizer='sgd',
    			loss='mse',
    			metrics=["accuracy"]
    			)

\#\#一、网络结构  
**保存网络结构和权值(save)**  
利用Model类的save函数，如

model.save('./model_name.h5')

**载入模型(load\_model)**

from keras.models import load_model
    model=load_model('./model_name.h5')

\#\#二、权值文件  
**保存权值文件(save\_weights)**

model.fit(input,output,epochs=100,batch_size=64)
    #权值的保存在训练完成之后
    model.save_weights('./weights.h5')

**载入权值文件（load\_weights）**

model=load_model('./model_name.h5')
    model.load_weights('./weights.h5')
    #载入后就可以预测了
    result=model.predict(input,batch_size=1)

发表评论取消回复

表情：

评论列表（有 0 条评论，156人围观）

还没有评论，来说两句吧...

相关阅读

相关 tf模型保存、载入

![1598479-20190918172729850-306340277.png][] ![1598479-20190918172739939-34480307.p

曾经终败给现在/ 2023年08月17日 16:07/ 0 赞/ 233 阅读

相关 tensorflow之pb文件保存与载入

pb是protocol（协议） buffer（缓冲）的缩写。TensorFlow训练模型后存成的pb文件，是一种表示模型（神经网络）结构的二进制文件，将图中的变量保存成为常量，

Love The Way You Lie/ 2023年06月30日 09:05/ 0 赞/ 59 阅读

相关 keras-搭建神经网络结构

搭建的流程： 1. import 2. train,test 数据 3. model=tf.keras.model.Sequential 4. model.compi

今天药忘吃喽~/ 2023年05月21日 15:26/ 0 赞/ 74 阅读

相关 DL之Keras：keras保存网络结构、网络拓扑图、网络模型(json、yaml、h5等)注意事项及代码实现

DL之Keras：keras保存网络结构、网络拓扑图、网络模型(json、yaml、h5等)注意事项及代码实现目录 keras保存网络结构、网络拓

迈不过友情╰/ 2022年12月23日 03:28/ 0 赞/ 166 阅读

相关 OpenCV Python --- 图像的载入、显示和保存

OpenCV Python --- 图像的载入、显示和保存本文是OpenCV 2 Computer Vision Application Programming Coo

偏执的太偏执、/ 2022年11月21日 14:53/ 0 赞/ 330 阅读

相关【Tensorflow+keras】使用keras API保存模型权重、plot画loss损失函数、保存训练loss值

举例实现（1）模型实现 import tensorflow as tf from tensorflow.keras.layers import

淡淡的烟草味﹌/ 2022年10月06日 15:53/ 0 赞/ 286 阅读

相关 keras 网络结构和权值文件的保存与载入

以lenet网络为例 from keras.models import Model from keras.layers import Input,Conv2D

小咪咪/ 2022年05月27日 01:07/ 0 赞/ 157 阅读

相关竞争网络权值分析

竞争网络权值分析 %竞争网络权值分析 clc; clear; P=\[0.7071 0.6402 0.000 -0.1961 0.1961 -0.9285 -0.8762 -

落日映苍穹つ/ 2022年05月16日 10:27/ 0 赞/ 245 阅读

相关 TensorFlow和Keras的模型保存及载入模型参数继续训练

TensorFlow 在TensorFlow中，模型的持久化保存和加载主要通过Saver()。在初次训练之后调用如下的save函数保存，然后，在预测前，或者在继续训

本是古典何须时尚/ 2022年02月13日 04:49/ 0 赞/ 880 阅读

相关 TensorFlow基础：网络结构的保存

1. 保存当前参数最好时的参数 saver_ckpt = tf.train.Saver(tf.trainable_variables())

迈不过友情╰/ 2022年01月28日 01:13/ 0 赞/ 224 阅读