使用Keras加载含有自定义层或函数的模型操作


Posted in Python onJune 10, 2020

当我们导入的模型含有自定义层或者自定义函数时,需要使用custom_objects来指定目标层或目标函数。

例如:

我的一个模型含有自定义层“SincConv1D”,需要使用下面的代码导入:

from keras.models import load_model
model = load_model('model.h5', custom_objects={'SincConv1D': SincConv1D})

如果不加custom_objects指定目标层Layer,则会出现以下报错:

ValueError: Unknown layer: SincConv1D

同样的,当我的模型含有自定义函数“my_loss”,需要使用下面的代码导入:

from keras.models import load_model
model = load_model('model.h5', custom_objects={'my_loss': my_loss})

补充知识:keras加载模型load_model报错——ValueError: Unknown layer: CRF

我就废话不多说了,大家还是直接看代码吧!

from keras.models import load_model
model = load_model(model_path)

会报错,需要在load_model函数中添加custom_objects参数,来声明自定义的层
(用keras搭建bilstm-crf,在训练模型时,使用的是:

from keras_contrib.layers.crf import CRF)

from keras_contrib.layers.crf import CRF, crf_loss, crf_viterbi_accuracy
model = load_model(model_path, custom_objects={"CRF": CRF, 'crf_loss': crf_loss,
                        'crf_viterbi_accuracy': crf_viterbi_accuracy})

以上这篇使用Keras加载含有自定义层或函数的模型操作就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
使用python将mdb数据库文件导入postgresql数据库示例
Feb 17 Python
Python程序中使用SQLAlchemy时出现乱码的解决方案
Apr 24 Python
Python常见加密模块用法分析【MD5,sha,crypt模块】
May 24 Python
PyQt5利用QPainter绘制各种图形的实例
Oct 19 Python
解决pycharm每次新建项目都要重新安装一些第三方库的问题
Jan 17 Python
关于python之字典的嵌套,递归调用方法
Jan 21 Python
Django多进程滚动日志问题解决方案
Dec 17 Python
根据tensor的名字获取变量的值方式
Jan 04 Python
浅谈SciPy中的optimize.minimize实现受限优化问题
Feb 29 Python
Python使用jpype模块调用jar包过程解析
Jul 29 Python
Python matplotlib多个子图绘制整合
Apr 13 Python
Python读取和写入Excel数据
Apr 20 Python
keras 获取某层的输入/输出 tensor 尺寸操作
Jun 10 #Python
Python 字典中的所有方法及用法
Jun 10 #Python
在keras 中获取张量 tensor 的维度大小实例
Jun 10 #Python
Keras—embedding嵌入层的用法详解
Jun 10 #Python
Keras框架中的epoch、bacth、batch size、iteration使用介绍
Jun 10 #Python
Python3.9 beta2版本发布了,看看这7个新的PEP都是什么
Jun 10 #Python
JAVA及PYTHON质数计算代码对比解析
Jun 10 #Python
You might like
基于empty函数的判断详解
2013/06/17 PHP
解析PHP实现下载文件的两种方法
2013/07/05 PHP
php实现统计网站在线人数的方法
2015/05/12 PHP
PHP单例模式应用示例【多次连接数据库只实例化一次】
2018/12/18 PHP
用jscript实现新建和保存一个word文档
2007/06/15 Javascript
JQuery UI皮肤定制
2009/07/27 Javascript
JavaScript学习历程和心得小结
2010/08/16 Javascript
《JavaScript高级程序设计》阅读笔记(三) ECMAScript中的引用类型
2012/02/27 Javascript
jQuery实现dialog设置focus焦点的方法
2015/06/10 Javascript
javascript创建对象、对象继承的实用方式详解
2016/03/08 Javascript
在JavaScript中对HTML进行反转义详解
2016/05/18 Javascript
微信小程序  简单实例(阅读器)的实例开发
2016/09/29 Javascript
Angular JS数据的双向绑定详解及实例
2016/12/31 Javascript
javascript表达式和运算符详解
2017/02/07 Javascript
完美实现js焦点轮播效果(一)
2017/03/07 Javascript
js实现音频控制进度条功能
2017/04/01 Javascript
bootstrap Table的一些小操作
2017/11/01 Javascript
前端MVVM框架解析之双向绑定
2018/01/24 Javascript
微信小程序模板消息推送的两种实现方式
2019/08/27 Javascript
JS几个常用的函数和对象定义与用法示例
2020/01/15 Javascript
解决vuex数据页面刷新后初始化操作
2020/07/26 Javascript
[04:45]上海特级锦标赛主赛事第三日TOP10
2016/03/05 DOTA
实例解析Python设计模式编程之桥接模式的运用
2016/03/02 Python
Python基于QRCode实现生成二维码的方法【下载,安装,调用等】
2017/07/11 Python
对Python 多线程统计所有csv文件的行数方法详解
2019/02/12 Python
python中如何实现将数据分成训练集与测试集的方法
2019/09/13 Python
Pytorch十九种损失函数的使用详解
2020/04/29 Python
python3 使用ssh隧道连接mysql的操作
2020/12/05 Python
联想美国官方商城:Lenovo美国
2017/06/19 全球购物
斯凯奇新西兰官网:SKECHERS新西兰
2018/02/22 全球购物
优秀员工自荐书
2013/12/19 职场文书
会计稽核岗位职责
2015/04/13 职场文书
红色经典电影观后感
2015/06/18 职场文书
2015年食品安全宣传周活动总结
2015/07/09 职场文书
PyCharm配置KBEngine快速处理代码提示冲突、配置命令问题
2021/04/03 Python
Python中相见恨晚的技巧
2021/04/13 Python