keras的load_model实现加载含有参数的自定义模型


Posted in Python onJune 22, 2020

网上的教程大多数是教大家如何加载自定义模型和函数,如下图

keras的load_model实现加载含有参数的自定义模型

这个SelfAttention层是在训练过程自己定义的一个class,但如果要加载这个自定义层,需要在load_model里添加custom_objects字典,这个自定义的类,不要用import ,最好是直接复制进再训练的模型中,这些是基本教程。

------------------分割线讲重点------------------

如果直接运行上面的代码,会出现一个init初始化错误,如下图,

keras的load_model实现加载含有参数的自定义模型

再来看看 这个SelfAttention 自定义的类的初始化

keras的load_model实现加载含有参数的自定义模型

这就说明再调用这个类的时候,输入的ch=256并不会初始化这个类,需要先自定义好初始化值,如下图

keras的load_model实现加载含有参数的自定义模型

调用方式不变

keras的load_model实现加载含有参数的自定义模型

这样问题就解决啦!

补充知识:keras load model的时候,报错('Keyword argument not understood:', u'******')如何解决

由于keras不同版本的API有变化,因此在一个keras版本下训练的模型在另一个keras版本下加载时,可能会出现诸如('Keyword argument not understood:', u'data_format')等报错。

通过打开*.h5文件,查看该模型训练所用环境,再安装该环境即可解决报错。

查看Keras Model所用的Keras环境的方法

import h5py

f = h5py.File('Model.h5', 'r')
print(f.attrs.get('keras_version'))

根据输出的keras版本安装对应版本的keras即可解决加载问题。

以上这篇keras的load_model实现加载含有参数的自定义模型就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python使用urllib模块和pyquery实现阿里巴巴排名查询
Jan 16 Python
python新手经常遇到的17个错误分析
Jul 30 Python
使用Python写一个小游戏
Apr 02 Python
在Pytorch中使用样本权重(sample_weight)的正确方法
Aug 17 Python
Python实现线性插值和三次样条插值的示例代码
Nov 13 Python
Django用户身份验证完成示例代码
Apr 03 Python
python中sklearn的pipeline模块实例详解
May 21 Python
浅谈pytorch 模型 .pt, .pth, .pkl的区别及模型保存方式
May 25 Python
浅析Python 序列化与反序列化
Aug 05 Python
python 绘制国旗的示例
Sep 27 Python
Python 中的函数装饰器和闭包详解
Feb 06 Python
python三子棋游戏
May 04 Python
python GUI模拟实现计算器
Jun 22 #Python
keras CNN卷积核可视化,热度图教程
Jun 22 #Python
python实现斗地主分牌洗牌
Jun 22 #Python
解决Keras使用GPU资源耗尽的问题
Jun 22 #Python
Keras - GPU ID 和显存占用设定步骤
Jun 22 #Python
Python 基于jwt实现认证机制流程解析
Jun 22 #Python
python中format函数如何使用
Jun 22 #Python
You might like
截获网站title标签之家内容的例子
2006/10/09 PHP
PHP中使用GD库创建圆形饼图的例子
2014/11/19 PHP
php截取指定2个字符之间字符串的方法
2015/04/15 PHP
在Mac OS上自行编译安装Apache服务器和PHP解释器
2015/12/24 PHP
php版微信公众平台实现预约提交后发送email的方法
2016/09/26 PHP
PHP类的特性实例分析
2016/09/28 PHP
PHP实现的折半查找算法示例
2017/12/19 PHP
PHP使用星号替代用户名手机和邮箱的实现代码
2018/02/07 PHP
JavaScript验证图片类型(扩展名)的函数分享
2014/05/05 Javascript
什么是Node.js?Node.js详细介绍
2014/06/01 Javascript
js中el表达式的使用和非空判断方法
2018/03/28 Javascript
React 无状态组件(Stateless Component) 与高阶组件
2018/08/14 Javascript
详解html-webpack-plugin插件(用法总结)
2018/09/12 Javascript
微信小程序实现九宫格抽奖
2020/04/15 Javascript
详解基于Vue/React项目的移动端适配方案
2019/08/23 Javascript
mapboxgl区划标签避让不遮盖实现的代码详解
2020/07/01 Javascript
[01:04:29]DOTA2-DPC中国联赛 正赛 Phoenix vs XG BO3 第二场 1月31日
2021/03/11 DOTA
python中使用smtplib和email模块发送邮件实例
2014/04/22 Python
Python中使用PyHook监听鼠标和键盘事件实例
2014/07/18 Python
使用Python的web.py框架实现类似Django的ORM查询的教程
2015/05/02 Python
Python中统计函数运行耗时的方法
2015/05/05 Python
教你使用python实现微信每天给女朋友说晚安
2018/03/23 Python
Python 通配符删除文件的实例
2018/04/24 Python
Tensorflow 查看变量的值方法
2018/06/14 Python
详解pandas.DataFrame.plot() 画图函数
2020/06/14 Python
Sublime Text3最新激活注册码分享适用2020最新版 亲测可用
2020/11/12 Python
css3 iphone玻璃透明气泡完美实现
2013/03/20 HTML / CSS
英国户外服装、鞋类和设备的领先零售商:Millets
2020/10/12 全球购物
关于工作经历的证明书
2014/10/11 职场文书
2014年评职称工作总结
2014/11/20 职场文书
校园运动会广播稿
2015/08/19 职场文书
怎样评估创业计划书是否有可行性?
2019/08/07 职场文书
导游词之泉州崇武古城
2019/12/20 职场文书
redis哨兵常用命令和监控示例详解
2021/05/27 Redis
python turtle绘图命令及案例
2021/11/23 Python
Window server 2012 R2 AD域的组策略相关设置
2022/04/28 Servers