使用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  连接字符串(join %)
Sep 06 Python
python统计文本文件内单词数量的方法
May 30 Python
Python常见加密模块用法分析【MD5,sha,crypt模块】
May 24 Python
pandas DataFrame 根据多列的值做判断,生成新的列值实例
May 18 Python
windows下pycharm安装、创建文件、配置默认模板
Jul 31 Python
Python对CSV、Excel、txt、dat文件的处理
Sep 18 Python
python创造虚拟环境方法总结
Mar 04 Python
详解python项目实战:模拟登陆CSDN
Apr 04 Python
Dlib+OpenCV深度学习人脸识别的方法示例
May 14 Python
用python做游戏的细节详解
Jun 25 Python
pandas实现excel中的数据透视表和Vlookup函数功能代码
Feb 14 Python
Pyspark获取并处理RDD数据代码实例
Mar 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 JSON格式数据交互实例代码详解
2011/01/13 PHP
php全排列递归算法代码
2012/10/09 PHP
解析thinkphp中的导入文件标签
2013/06/20 PHP
thinkphp5 加载静态资源路径与常量的方法
2017/12/24 PHP
浅析PHP中的闭包和匿名函数
2017/12/25 PHP
PHP性能分析工具xhprof的安装使用与注意事项
2017/12/19 PHP
JavaScript入门教程(8) Location地址对象
2009/01/31 Javascript
基于JQUERY的两个ListBox子项互相调整的实现代码
2011/05/07 Javascript
nodejs教程之环境安装及运行
2014/11/21 NodeJs
JavaScript获取当前网页最后修改时间的方法
2015/04/03 Javascript
JavaScript获取ul中li个数的方法
2017/02/13 Javascript
JS实现移动端按首字母检索城市列表附源码下载
2017/07/05 Javascript
React Native第三方平台分享的实例(Android,IOS双平台)
2017/08/04 Javascript
Node.Js生成比特币地址代码解析
2018/04/21 Javascript
使用Vue开发自己的Chrome扩展程序过程详解
2019/06/21 Javascript
Vue中多元素过渡特效的解决方案
2020/02/05 Javascript
jQuery zTree如何改变指定节点文本样式
2020/10/16 jQuery
Node快速切换版本、版本回退(降级)、版本更新(升级)
2021/01/07 Javascript
python测试驱动开发实例
2014/10/08 Python
Python配置mysql的教程(推荐)
2017/10/13 Python
python字符串常用方法
2018/06/14 Python
python 统计列表中不同元素的数量方法
2018/06/29 Python
33个Python爬虫项目实战(推荐)
2019/07/08 Python
印尼太阳百货公司网站:Matahari
2018/02/04 全球购物
工商管理专业实习大学生自我鉴定
2013/09/19 职场文书
公益活动策划方案
2014/01/09 职场文书
大学总结自我鉴定
2014/01/18 职场文书
人力资源管理专业自荐书范文
2014/02/10 职场文书
企业文化理念标语
2014/06/10 职场文书
销售人才自我评价范文
2014/09/27 职场文书
投标人法定代表人授权委托书格式
2014/09/28 职场文书
2015驻村干部工作总结
2015/04/07 职场文书
难以忽视的真相观后感
2015/06/05 职场文书
小学二年级班主任工作经验交流材料
2015/11/02 职场文书
pandas提升计算效率的一些方法汇总
2021/05/30 Python
Pandas 数据编码的十种方法
2022/04/20 Python