使用Keras加载含有自定义层或函数的模型操作
更新时间：2020年06月10日 11:26:56 作者：z小白
这篇文章主要介绍了使用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加载含有自定义层或函数的模型操作就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了如何修改numpy array的数据类型，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-05-05
这篇文章将详细讲解图像灰度非线性变换。图像灰度非线性变换主要包括对数变换、幂次变换、指数变换、分段函数变换，通过非线性关系对图像进行灰度处理，本文主要讲解三种常见类型的灰度非线性变换，感兴趣的可以了解一下2022-03-03
这篇文章主要介绍了Python实现周期性抓取网页内容的方法,涉及Python时间函数及正则匹配的相关操作技巧,具有一定参考借鉴价值,需要的朋友可以参考下2015-11-11
很多网站登录登陆时都要用到滑块验证码，本文主要介绍了selenium+opencv实现滑块验证码的登陆，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-04-04
这篇文章主要介绍了关于python 的legend图例,参数使用说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-04-04
今天小编就为大家分享一篇解决python字典对值(值为列表)赋值出现重复的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
Django中在xadmin中集成DjangoUeditor过程详解
这篇文章主要介绍了Django中在xadmin中集成DjangoUeditor过程详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07
在本篇文章里小编给大家整理的是一篇关于python析构函数用法及注意事项，有需要的朋友们可以学习参考下。2021-06-06
这篇文章主要介绍了Django和Flask框架相关对比,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-10-10
最近有粉丝向我咨询super相关的问题，说网上搜索到的教程不够通俗易懂，看了之后还是不太理解。所以在这里基于我自己的理解来讲解一下super2021-10-10
最新评论