使用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 相关文章推荐
使用wxPython获取系统剪贴板中的数据的教程
May 06 Python
详解Python pygame安装过程笔记
Jun 05 Python
Python爬虫实例爬取网站搞笑段子
Nov 08 Python
python解析html提取数据,并生成word文档实例解析
Jan 22 Python
利用Anaconda简单安装scrapy框架的方法
Jun 13 Python
python批量图片处理简单示例
Aug 06 Python
Django和Flask框架优缺点对比
Oct 24 Python
Python进程池Pool应用实例分析
Nov 27 Python
python range实例用法分享
Feb 06 Python
详解Anaconda 的安装教程
Sep 23 Python
python 下载文件的几种方法汇总
Jan 06 Python
OpenCV-Python直方图均衡化实现图像去雾
Jun 07 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
实用函数10
2007/11/08 PHP
mysql_num_rows VS COUNT 效率问题分析
2011/04/23 PHP
将word转化为swf 如同百度文库般阅读实现思路及代码
2013/08/09 PHP
php文件缓存类用法实例分析
2015/04/22 PHP
PHP检测数据类型的几种方法(总结)
2017/03/04 PHP
PHP计算近1年的所有月份
2017/03/13 PHP
浅谈PHP错误类型及屏蔽方法
2017/05/27 PHP
PHP实现登录注册之BootStrap表单功能
2017/09/03 PHP
下载站控制介绍字数显示的脚本 显示全部 隐藏介绍等功能
2009/09/19 Javascript
JavaScript 监听textarea中按键事件
2009/10/08 Javascript
JQuery 文本框使用小结
2010/05/22 Javascript
屏蔽IE弹出"您查看的网页正在试图关闭窗口,是否关闭此窗口"的方法
2013/12/31 Javascript
按Enter键触发事件的jquery方法实现代码
2014/02/17 Javascript
基于promise.js实现nodejs的promises库
2014/07/06 NodeJs
JavaScript DOM节点添加示例
2014/07/16 Javascript
javascript实现获取浏览器版本、操作系统类型
2015/01/29 Javascript
JS+CSS实现六级网站导航主菜单效果
2015/09/28 Javascript
javascript实现无法关闭的弹框
2016/11/27 Javascript
使用prop解决一个checkbox选中后再次选中失效的问题
2017/07/05 Javascript
Angular4的输入属性与输出属性实例详解
2017/11/29 Javascript
vue.js中created方法作用
2018/03/30 Javascript
vue-for循环嵌套操作示例
2019/01/28 Javascript
深入学习JavaScript 高阶函数
2019/06/11 Javascript
jQuery擦除插件eraser使用方法详解
2020/01/11 jQuery
[42:20]Winstrike vs VGJ.S 2018国际邀请赛淘汰赛BO3 第二场 8.23
2018/08/24 DOTA
Python+Appium实现自动化测试的使用步骤
2020/03/24 Python
基于OpenCV的路面质量检测的实现
2020/11/04 Python
5分钟让你掌握css3阴影、倒影、渐变小技巧(小编推荐)
2016/08/15 HTML / CSS
html5实现canvas阴影效果示例
2014/05/07 HTML / CSS
如何写出好的Java代码
2014/04/25 面试题
大学生自助营养快餐店创业计划书
2014/01/13 职场文书
小学运动会广播稿200字(十二篇)
2014/01/14 职场文书
建筑系毕业生自我鉴定
2014/01/24 职场文书
和谐家庭演讲稿
2014/05/24 职场文书
活动新闻稿范文
2015/07/17 职场文书
Python还能这么玩之用Python修改了班花的开机密码
2021/06/04 Python