使用Keras加载含有自定义层或函数的模型操作
作者:z小白 时间:2022-12-25 19:29:29
当我们导入的模型含有自定义层或者自定义函数时,需要使用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})
来源:https://blog.csdn.net/zzc15806/article/details/88222125
标签:Keras,加载,自定义层,函数
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
Python字符串逆序输出的实例讲解
2023-04-21 22:59:46
FrontPage XP设计教程2——网页的编辑
2008-10-11 12:16:00
![](https://img.aspxhome.com/file/UploadPic/200810/17/20081017121938712s.jpg)
mysql中的int(10)int(20)分别代表什么意思
2024-01-20 21:31:30
Python实现的旋转数组功能算法示例
2021-09-11 20:49:46
采用memcache在web集群中实现session的同步会话
2023-10-01 22:26:37
Mootools 1.2教程(21)——类(二)
2008-12-28 20:58:00
利于python脚本编写可视化nmap和masscan的方法
2021-04-14 11:19:52
![](https://img.aspxhome.com/file/2023/9/121009_0s.jpg)
Python求离散序列导数的示例
2022-10-04 11:18:52
![](https://img.aspxhome.com/file/2023/7/63987_0s.jpg)
vscode安装rainbow-fart(鼓励师)插件
2022-05-09 14:11:42
![](https://img.aspxhome.com/file/2023/2/132272_0s.jpg)
Django REST framework 如何实现内置访问频率控制
2023-08-01 17:11:00
Python如何对齐字符串
2023-05-30 01:21:11
python实现简单文件读写函数
2023-08-29 04:37:24
python判断一个集合是否包含了另外一个集合中所有项的方法
2022-01-15 19:08:06
网易首页的新闻代码
2022-02-24 16:01:15
python实现弹窗祝福效果
2021-09-08 04:22:15
![](https://img.aspxhome.com/file/2023/2/70792_0s.jpg)
Python标准库之循环器(itertools)介绍
2023-09-16 02:41:43
HTTP Headers 傻瓜教程[译]
2010-01-25 12:48:00
![](/file/UploadPic/20101/25/http_diagram-83s.png)
Python 通配符删除文件的实例
2022-04-24 03:00:22
去除DW MX 2004表格宽度辅助
2010-09-02 12:37:00
![](https://img.aspxhome.com/file/UploadPic/20072/200723113022142s.gif)
Windows10下mysql 8.0.19 winx64安装教程及修改初始密码
2024-01-26 11:03:37
![](https://img.aspxhome.com/file/2023/3/70463_0s.jpg)