IT编程 > 脚本编程 > Python

使用Keras加载含有自定义层或函数的模型操作

108883人参与2020-06-18

当我们导入的模型含有自定义层或者自定义函数时,需要使用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加载含有自定义层或函数的模型操作就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持萬仟网。

您对本文有任何疑问!!点此进行留言回复

推荐阅读

猜你喜欢

使用Keras加载含有自定义层或函数的模型操作

06-18

keras 获取某层的输入/输出 tensor 尺寸操作

06-18

keras 解决加载lstm+crf模型出错的问题

06-18

Python 字典中的所有方法及用法

06-18

浅谈keras中的目标函数和优化函数MSE用法

06-18

Keras自定义IOU方式

06-18

拓展阅读

大家都在看

Numpy用于数据存储和读取

12-06

python 异常处理 StopIteration 用来作为迭代器的输出停止/next()

07-12

OpenCV 画多边形 — cv.polylines()函数使用

10-06

ROS笔记——在conda的python3.7中使用ROS Kinetic

07-30

使用Keras加载含有自定义层或函数的模型操作

06-18

python 实现rolling和apply函数的向下取值操作

06-23

pycharm 2020 Error: failed to send plot to http://127.0.0.1:63342

07-30

使用Python来开发Markdown脚本扩展的实例分享

05-30

热门评论