使用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 默认参数问题的陷阱
Feb 29 Python
视觉直观感受若干常用排序算法
Apr 13 Python
Python SqlAlchemy动态添加数据表字段实例解析
Feb 07 Python
对python中执行DOS命令的3种方法总结
May 12 Python
python list格式数据excel导出方法
Oct 31 Python
值得收藏,Python 开发中的高级技巧
Nov 23 Python
Python操作redis实例小结【String、Hash、List、Set等】
May 16 Python
基于Python新建用户并产生随机密码过程解析
Oct 08 Python
Python 生成器,迭代,yield关键字,send()传参给yield语句操作示例
Oct 12 Python
Python3如何使用tabulate打印数据
Sep 25 Python
详解使用python爬取抖音app视频(appium可以操控手机)
Jan 26 Python
python numpy中multiply与*及matul 的区别说明
May 26 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
PHP无限分类的类
2007/01/02 PHP
PHP与SQL注入攻击[二]
2007/04/17 PHP
一个php导出oracle库的php代码
2009/04/20 PHP
php Rename 更改文件、文件夹名称
2011/05/24 PHP
PHP 杂谈《重构-改善既有代码的设计》之五 简化函数调用
2012/05/07 PHP
基于ubuntu下nginx+php+mysql安装配置的具体操作步骤
2013/04/28 PHP
php源码 fsockopen获取网页内容实例详解
2016/09/24 PHP
总结PHP内存释放以及垃圾回收
2018/03/29 PHP
PHP实现单条sql执行多个数据的insert语句方法
2019/10/11 PHP
javasript实现密码的隐藏与显示
2015/05/08 Javascript
浅谈javascript的call()、apply()、bind()的用法
2016/02/21 Javascript
JavaScript 函数的执行过程
2016/05/09 Javascript
JS获取元素多层嵌套思路详解
2016/05/16 Javascript
Javascript oop设计模式 面向对象编程简单实例介绍
2016/12/13 Javascript
JS基于递归算法实现1,2,3,4,5,6,7,8,9倒序放入数组中的方法
2017/01/03 Javascript
Vue.js中关于侦听器(watch)的高级用法示例
2018/05/02 Javascript
element-ui表格数据转换的示例代码
2018/08/24 Javascript
VUE v-for循环中每个item节点动态绑定不同函数的实例
2018/09/26 Javascript
js实现窗口全屏示例详解
2019/09/17 Javascript
你不可不知的Vue.js列表渲染详解
2019/10/01 Javascript
three.js利用卷积法如何实现物体描边效果
2019/11/27 Javascript
vue学习笔记之Vue中css动画原理简单示例
2020/02/29 Javascript
浅谈vue websocket nodeJS 进行实时通信踩到的坑
2020/09/22 NodeJs
用Nodejs实现在终端中炒股的实现
2020/10/18 NodeJs
在vue中获取wangeditor的html和text的操作
2020/10/23 Javascript
[18:16]sakonoko 2017年卡尔集锦
2018/02/06 DOTA
数据挖掘之Apriori算法详解和Python实现代码分享
2014/11/07 Python
详解python列表生成式和列表生成式器区别
2019/03/27 Python
PyQt5实现让QScrollArea支持鼠标拖动的操作方法
2019/06/19 Python
详解python pandas 分组统计的方法
2019/07/30 Python
Python算法中的时间复杂度问题
2019/11/19 Python
“型”走纽约上东区:Sam Edelman
2017/04/02 全球购物
小学教育毕业生自荐信
2013/11/18 职场文书
医院党员公开承诺书
2014/08/30 职场文书
小学校长开学致辞
2015/07/29 职场文书
Python 中数组和数字相乘时的注意事项说明
2021/05/10 Python