使用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使用rabbitmq实现网络爬虫示例
Feb 20 Python
Python处理RSS、ATOM模块FEEDPARSER介绍
Feb 18 Python
Python函数式编程指南(二):从函数开始
Jun 24 Python
python转换字符串为摩尔斯电码的方法
Jul 06 Python
Python实现将一个大文件按段落分隔为多个小文件的简单操作方法
Apr 17 Python
Python 实现淘宝秒杀的示例代码
Jan 02 Python
Python元组及文件核心对象类型详解
Feb 11 Python
查看Django和flask版本的方法
May 14 Python
Python利用FFT进行简单滤波的实现
Feb 26 Python
PIP和conda 更换国内安装源的方法步骤
Sep 21 Python
让文件路径提取变得更简单的Python Path库
May 27 Python
Python基础学习之奇异的GUI对话框
May 27 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函数实现判断是否移动端访问
2015/03/03 PHP
标准PHP的AES加密算法类
2015/03/12 PHP
基于php的CMS中展示文章类实例分析
2015/06/18 PHP
php微信公众平台交互与接口详解
2016/11/28 PHP
PHP的PDO事务与自动提交
2019/01/24 PHP
laravel 5.5 关闭token的3种实现方式
2019/10/24 PHP
鼠标拖动实现DIV排序示例代码
2013/10/14 Javascript
JavaScript伸缩的菜单简单示例
2013/12/03 Javascript
JavaScript关闭当前页面(窗口)不带任何提示
2014/03/26 Javascript
js获取微信版本号的方法
2015/05/12 Javascript
javascript每日必学之条件分支
2016/02/17 Javascript
JS事件添加和移出的兼容写法示例
2016/06/20 Javascript
微信jssdk用法汇总
2016/07/16 Javascript
bootstrap table插件的分页与checkbox使用详解
2017/07/23 Javascript
Vue中之nextTick函数源码分析详解
2017/10/17 Javascript
深入理解JavaScript 中的执行上下文和执行栈
2018/10/23 Javascript
js模拟F11页面全屏显示
2019/09/17 Javascript
python以环状形式组合排列图片并输出的方法
2015/03/17 Python
python交互式图形编程实例(三)
2017/11/17 Python
tensorflow入门之训练简单的神经网络方法
2018/02/26 Python
Django添加sitemap的方法示例
2018/08/06 Python
Python爬虫实现验证码登录代码实例
2019/05/10 Python
python 求某条线上特定x值或y值的点坐标方法
2019/07/09 Python
Python semaphore evevt生产者消费者模型原理解析
2020/03/18 Python
python绘制汉诺塔
2021/03/01 Python
比利时买床:Beter Bed
2017/12/06 全球购物
英国建筑用品在线:Building Supplies Online(BSO)
2018/04/30 全球购物
伦敦的高级牛仔布专家:Trilogy
2018/08/06 全球购物
大学军训感言1000字
2014/02/25 职场文书
招聘专员岗位职责
2014/03/07 职场文书
舞蹈教育学专业自荐信
2014/06/15 职场文书
2015年秋季运动会加油稿
2015/07/22 职场文书
退休教师欢送会致辞
2015/07/31 职场文书
中考百日冲刺决心书
2015/09/22 职场文书
详解TypeScript中的类型保护
2021/04/29 Javascript
HTML+CSS实现导航条下拉菜单的示例代码
2021/08/02 HTML / CSS