基于keras 模型、结构、权重保存的实现


Posted in Python onJanuary 24, 2020

如何将训练好的网络进行保存,我们可以用pickle或cPickle来保存Keras模型,同时我们可以用下面的方法:

一、保存整个模型

model.save(filepath)将Keras模型和权重保存在一个HDF5文件中,该文件将包含:

模型的结构

模型的权重

训练配置(损失函数,优化器,准确率等)

优化器的状态,以便于从上次训练中断的地方

前提是已经安装python的h5py包.

from keras.models import load_model

当我们再一次使用时可以model.load_model(filepath)载入模型

二、保存模型的结构

model.to_jason()将模型序列化保存为json文件,里面记录了网络的整体结构, 各个层的参数设置等信息. 将json字符串保存到文件.

open(‘filename.json','w').write(json_string)
from keras.models import model_form_json
json_string=open('filename.json').read()

model=model_from_json(json_string)

除了json格式,还可以保存为yaml格式的字符串,形式与JSON一样

三、保存模型权重

model.save_weights()

我们经过调参后网络的输出精度比较满意后,可以将训练好的网络权重参数保存下 来.可通过下面的代码利用HDF5进行保存

model.save_weights(‘model_weights.h5')

使用的时加载模型:

model.load_weights(‘model_weights.h5')

如果你需要加载权重到不同的网络结构(有些层一样)中,例如fine-tune或transfer-learning,你可以通过层名字来加载模型:

model.load_weights('my_model_weights.h5', by_name=True)

因此我们建模时最好给每个层定义名字

以上就是我们保存模型的三种方法,需要我们在实践时多总结。

这篇基于keras 模型、结构、权重保存的实现就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python解析html开发库pyquery使用方法
Feb 07 Python
Python入门之modf()方法的使用
May 15 Python
在Django的模型和公用函数中使用惰性翻译对象
Jul 27 Python
python实现Adapter模式实例代码
Feb 09 Python
面向初学者的Python编辑器Mu
Oct 08 Python
对Python中for复合语句的使用示例讲解
Nov 01 Python
python用win32gui遍历窗口并设置窗口位置的方法
Jul 26 Python
对Django的restful用法详解(自带的增删改查)
Aug 28 Python
IronPython连接MySQL的方法步骤
Dec 27 Python
使用python处理题库表格并转化为word形式的实现
Apr 14 Python
python 怎样进行内存管理
Nov 10 Python
基于Python实现射击小游戏的制作
Apr 06 Python
Python 文件数据读写的具体实现
Jan 24 #Python
利用keras加载训练好的.H5文件,并实现预测图片
Jan 24 #Python
keras模型可视化,层可视化及kernel可视化实例
Jan 24 #Python
keras 特征图可视化实例(中间层)
Jan 24 #Python
基于keras输出中间层结果的2种实现方式
Jan 24 #Python
tensorflow 保存模型和取出中间权重例子
Jan 24 #Python
tensorflow 模型权重导出实例
Jan 24 #Python
You might like
ThinkPHP之foreach标签使用概述
2014/06/30 PHP
phpmyadmin中禁止外网使用的方法
2014/11/04 PHP
Yii2中如何使用modal弹窗(基本使用)
2016/05/30 PHP
jQuery中与toggleClass等价的程序段 以及未来学习的方向
2010/03/18 Javascript
在表单提交前进行验证的几种方式整理
2013/07/31 Javascript
举例详解JavaScript中Promise的使用
2015/06/24 Javascript
Javascript类型转换的规则实例解析
2016/02/23 Javascript
AngularJS中$http服务常用的应用及参数
2016/08/22 Javascript
javascript使用btoa和atob来进行Base64转码和解码
2017/03/20 Javascript
Ionic3 UI组件之autocomplete详解
2017/06/08 Javascript
Bootstrap Table使用整理(一)
2017/06/09 Javascript
linux 后台运行node服务指令方法
2018/05/23 Javascript
Vuejs2 + Webpack框架里,模拟下载的实例讲解
2018/09/05 Javascript
Vue 中 a标签上href无法跳转的解决方式
2019/11/12 Javascript
Python读取图片属性信息的实现方法
2016/09/11 Python
关于python pyqt5安装失败问题的解决方法
2017/08/08 Python
python编程实现随机生成多个椭圆实例代码
2018/01/03 Python
python 找出list中最大或者最小几个数的索引方法
2018/10/30 Python
为何人工智能(AI)首选Python?读完这篇文章你就知道了(推荐)
2019/04/06 Python
python opencv 简单阈值算法的实现
2019/08/04 Python
Python传递参数的多种方式(小结)
2019/09/18 Python
Python for循环及基础用法详解
2019/11/08 Python
pandas factorize实现将字符串特征转化为数字特征
2019/12/19 Python
python 基于opencv 绘制图像轮廓
2020/12/11 Python
Melijoe时尚童装德国官网:Melijoe德国
2016/09/03 全球购物
StubHub墨西哥:购买和出售您的门票
2016/09/17 全球购物
世界上最大的街头服饰网站:Karmaloop
2017/02/04 全球购物
什么是继承
2013/12/07 面试题
酒店保安员岗位职责
2014/01/31 职场文书
学生喝酒检讨书
2014/02/06 职场文书
国培计划培训感言
2014/03/11 职场文书
项目申报专员岗位职责
2014/07/09 职场文书
2015年店长工作总结范文
2015/04/08 职场文书
2016年春季运动会广播稿
2015/08/19 职场文书
MySQL8.0 Undo Tablespace管理详解
2022/06/16 MySQL
Win11远程连接不上怎么办?Win11远程桌面用不了的解决方法
2022/08/05 数码科技