使用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 相关文章推荐
Python3基础之函数用法
Aug 13 Python
在Django中编写模版节点及注册标签的方法
Jul 20 Python
python图像常规操作
Nov 11 Python
Python中pygal绘制雷达图代码分享
Dec 07 Python
Python推导式简单示例【列表推导式、字典推导式与集合推导式】
Dec 04 Python
python处理两种分隔符的数据集方法
Dec 12 Python
pandas DataFrame 删除重复的行的实现方法
Jan 29 Python
Python实现计算文件MD5和SHA1的方法示例
Jun 11 Python
python代理工具mitmproxy使用指南
Jul 04 Python
关于阿里云oss获取sts凭证 app直传 python的实例
Aug 20 Python
Python实现PyPDF2处理PDF文件的方法示例
Sep 25 Python
Python内置的数据类型及使用方法
Apr 13 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
总集篇&特番节目先行播出!《SAO Alicization War of Underworld》第2季度TV动画4月25日放送!
2020/03/06 日漫
javascript encodeURI和encodeURIComponent的比较
2010/04/03 Javascript
ExtJS PropertyGrid中使用Combobox选择值问题
2010/06/13 Javascript
用js实现table单元格高宽调整,兼容合并单元格(兼容IE6、7、8、FF)实例
2013/06/25 Javascript
nodejs npm install全局安装和本地安装的区别
2014/06/05 NodeJs
node.js中的console.time方法使用说明
2014/12/09 Javascript
jQuery插件jRumble实现网页元素抖动
2015/06/05 Javascript
使用JQuery在线制作ppt并在线演示源码特效
2015/09/08 Javascript
值得分享的Bootstrap Ace模板实现菜单和Tab页效果
2015/12/30 Javascript
jquery插件jquery.dragscale.js实现拖拽改变元素大小的方法(附demo源码下载)
2016/02/25 Javascript
jQuery javascript获得网页的高度与宽度的实现代码
2016/04/26 Javascript
JS实现获取剪贴板内容的方法
2016/06/21 Javascript
AngularJS之依赖注入模拟实现
2016/08/19 Javascript
微信小程序 location API实例详解
2016/10/02 Javascript
js仿微信抢红包功能
2020/09/25 Javascript
基于js文件加载优化(详解)
2018/01/03 Javascript
学习Vue组件实例
2018/04/28 Javascript
vue用BMap百度地图实现即时搜索功能
2019/09/26 Javascript
Vue.js组件通信之自定义事件详解
2019/10/19 Javascript
openLayer4实现动态改变标注图标
2020/08/17 Javascript
[01:28:31]《加油DOTA》真人秀 第五期
2014/09/01 DOTA
Python scikit-learn 做线性回归的示例代码
2017/11/01 Python
python定向爬取淘宝商品价格
2018/02/27 Python
python使用pygame实现笑脸乒乓球弹珠球游戏
2019/11/25 Python
解决Pycharm中恢复被exclude的项目问题(pycharm source root)
2020/02/14 Python
如何用Python绘制3D柱形图
2020/09/16 Python
Python用dilb提取照片上人脸的示例
2020/10/26 Python
plt.figure()参数使用详解及运行演示
2021/01/08 Python
Nike德国官网:Nike.com (DE)
2018/11/13 全球购物
华美博弈C/VC工程师笔试试题
2012/07/16 面试题
小学教师岗位职责
2013/11/25 职场文书
幼儿园门卫制度
2014/01/29 职场文书
小学生竞选班干部演讲稿
2014/04/24 职场文书
警察先进个人事迹材料
2014/05/16 职场文书
公司应聘自荐书
2014/06/14 职场文书
客户付款通知书
2015/04/23 职场文书