使用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基础教程之类class定义使用方法
Feb 20 Python
python获取网页状态码示例
Mar 30 Python
python使用any判断一个对象是否为空的方法
Nov 19 Python
浅谈MySQL中的触发器
May 05 Python
python更新列表的方法
Jul 28 Python
利用Python爬取微博数据生成词云图片实例代码
Aug 31 Python
Python 实现Windows开机运行某软件的方法
Oct 14 Python
python实现大转盘抽奖效果
Jan 22 Python
Python基于模块Paramiko实现SSHv2协议
Apr 28 Python
pyqt5实现井字棋的示例代码
Dec 07 Python
Python控制鼠标键盘代码实例
Dec 08 Python
pytorch 6 batch_train 批训练操作
May 28 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
咖啡店都有些什么常规豆子呢?有什么风味在里面
2021/03/04 咖啡文化
Windows下Apache + PHP SESSION丢失的解决过程全纪录
2015/04/07 PHP
使用Appcan客户端自动更新PHP版本号(全)
2015/07/31 PHP
PHP实现对文件锁进行加锁、解锁操作的方法
2017/07/04 PHP
详解cookie验证的php应用的一种SSO解决办法
2017/10/20 PHP
yii2多图上传组件的使用教程
2018/05/10 PHP
laravel 解决ajax异步提交数据,并还回填充表格的问题
2019/10/15 PHP
用jscript实现列出安装的软件列表
2007/06/18 Javascript
js constructor的实际作用分析
2011/11/15 Javascript
js的Boolean对象初始值示例
2014/03/04 Javascript
javascript学习笔记(五)原型和原型链详解
2014/10/08 Javascript
JavaScript获取DOM元素的11种方法总结
2015/04/25 Javascript
详解JavaScript中localStorage使用要点
2016/01/13 Javascript
Bootstrap模块dropdown实现下拉框响应
2016/05/22 Javascript
值得分享和收藏的xmlplus组件学习教程
2017/05/05 Javascript
在vue-cli脚手架中配置一个vue-router前端路由
2017/07/03 Javascript
jQuery实现的下雪动画效果示例【附源码下载】
2018/02/02 jQuery
父组件中vuex方法更新state子组件不能及时更新并渲染的完美解决方法
2018/04/25 Javascript
微信小程序点击保存图片到本机功能
2019/12/13 Javascript
js事件机制----捕获与冒泡机制实例分析
2020/05/22 Javascript
解决windows下Sublime Text 2 运行 PyQt 不显示的方法分享
2014/06/18 Python
python实现无证书加密解密实例
2014/10/27 Python
python爬虫实现教程转换成 PDF 电子书
2017/02/19 Python
浅谈Python中range和xrange的区别
2017/12/20 Python
Django使用httpresponse返回用户头像实例代码
2018/01/26 Python
Python装饰器用法实例总结
2018/05/26 Python
python实现维吉尼亚算法
2019/03/20 Python
python实现微信防撤回神器
2019/04/29 Python
Python Pandas 如何shuffle(打乱)数据
2019/07/30 Python
C/C++程序员常见面试题一
2012/12/08 面试题
2014年煤矿工作总结
2014/11/24 职场文书
标准演讲稿格式结尾应该怎么书写?
2019/07/17 职场文书
创业计划书之冷饮店
2019/09/27 职场文书
如何用python绘制雷达图
2021/04/24 Python
详解Python如何批量采集京东商品数据流程
2022/01/22 Python
spring注解 @PropertySource配置数据源全流程
2022/03/25 Java/Android