使用Keras加载含有自定义层或函数的模型操作
当我们导入的模型含有自定义层或者自定义函数时，需要使用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中如何实现真正的按位取反运算的相关资料,需要的朋友可以参考下2023-02-02
在本篇文章里小编给大家整理的是一篇关于python通过opencv调用摄像头操作实例分析内容，有兴趣的朋友们可以学习下。2021-06-06
这篇文章主要介绍了python基于concurrent模块实现多线程，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下2021-04-04
下面小编就为大家分享一篇python爬虫 使用真实浏览器打开网页的两种方法总结，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-04-04
生成器指的是生成器对象,可以由生成器表达式得到,也可以使用yield关键字得到一个生成器函数,调用这个函数得到一个生成器对象,下面这篇文章主要给大家介绍了关于Python中生成器的相关资料,需要的朋友可以参考下2022-04-04
这篇文章主要介绍了SciPy中两个模块:io 和misc的使用，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-05-05
Pytorch BCELoss和BCEWithLogitsLoss的使用
这篇文章主要介绍了Pytorch BCELoss和BCEWithLogitsLoss的使用详解，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-05-05
django 将自带的数据库sqlite3改成mysql实例
这篇文章主要介绍了django 将自带的数据库sqlite3改成mysql实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-07-07
python中的运算符主要包括算术运算符，关系(比较)运算符，赋值运算符，逻辑运算符，成员运算符，身份运算符，三目运算符。使用运算符将不同类型的数据按照一定的规则连接起来的式子，称为表达式。下面将介绍一些常用的运算符2022-08-08
协程是一种用户态的轻量级线程，又称微线程。这篇文章主要介绍了python编程使用协程并发的优缺点，感兴趣的朋友跟随小编一起看看吧2018-09-09
最新评论