使用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 相关文章推荐
centos系统升级python 2.7.3
Jul 03 Python
Python 多线程的实例详解
Sep 07 Python
Python3实现带附件的定时发送邮件功能
Dec 22 Python
Python简单获取网卡名称及其IP地址的方法【基于psutil模块】
May 24 Python
python查看模块安装位置的方法
Oct 16 Python
Django中的cookie和session
Aug 27 Python
tensorflow 实现数据类型转换
Feb 17 Python
10个python3常用排序算法详细说明与实例(快速排序,冒泡排序,桶排序,基数排序,堆排序,希尔排序,归并排序,计数排序)
Mar 17 Python
Python pandas 列转行操作详解(类似hive中explode方法)
May 18 Python
使用python修改文件并立即写回到原始位置操作(inplace读写)
Jun 28 Python
pyqt5实现井字棋的示例代码
Dec 07 Python
Pytorch反向传播中的细节-计算梯度时的默认累加操作
Jun 05 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 - Html Transfer Code
2006/10/09 PHP
PHP笔记之:日期函数的使用介绍
2013/04/24 PHP
php根据分类合并数组的方法实例详解
2013/11/06 PHP
php compact 通过变量创建数组
2016/11/15 PHP
jquery tools 系列 scrollable学习
2009/09/06 Javascript
JQuery.uploadify 上传文件插件的使用详解 for ASP.NET
2010/01/22 Javascript
JavaScript实现的图像模糊算法代码分享
2014/04/22 Javascript
js修改onclick动作的四种方法(推荐)
2016/08/18 Javascript
总结Javascript中的隐式类型转换
2016/08/24 Javascript
JS对大量数据进行多重过滤的方法
2016/11/04 Javascript
Nodejs进阶:如何将图片转成datauri嵌入到网页中去实例
2016/11/21 NodeJs
Angularjs之filter过滤器(推荐)
2016/11/27 Javascript
JS实现的简单图片切换功能示例【测试可用】
2017/02/14 Javascript
js中作用域的实例解析
2017/03/16 Javascript
React Native使用百度Echarts显示图表的示例代码
2017/11/07 Javascript
js解决软键盘遮挡输入框的问题分享
2017/12/19 Javascript
vue实现多个元素或多个组件之间动画效果
2018/09/25 Javascript
vue使用rem实现 移动端屏幕适配
2018/09/26 Javascript
详解在React-Native中持久化redux数据
2019/05/22 Javascript
jquery使用echarts实现有向图可视化功能示例
2019/11/25 jQuery
js键盘事件实现人物的行走
2020/01/17 Javascript
[02:30]辉夜杯主赛事第二日胜者组半决赛 CDEC.Y赛后采访
2015/12/26 DOTA
用Python进行一些简单的自然语言处理的教程
2015/03/31 Python
Python 学习教程之networkx
2019/04/15 Python
如何在Cloud Studio上执行Python代码?
2019/08/09 Python
pygame实现飞机大战
2020/03/11 Python
Django框架安装及项目创建过程解析
2020/09/14 Python
AmazeUI 网格的实现示例
2020/08/13 HTML / CSS
国际知名设计师时装商店:Coggles
2016/09/05 全球购物
美国电子产品购物网站:BuyDig.com
2020/06/17 全球购物
企业党员一句话承诺
2014/05/30 职场文书
乡文化站暑期培训方案
2014/08/28 职场文书
皇城相府导游词
2015/02/06 职场文书
2016年十一促销广告语
2016/01/28 职场文书
python中%格式表达式实例用法
2021/06/18 Python
MySql数据库 查询时间序列间隔
2022/05/11 MySQL