硬汉女主角,用excel制作表格,sswyt
当我们导入的模型含有自定义层或者自定义函数时,需要使用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 实现将numpy中的nan和inf,nan替换成对应的均值
python爬虫把url链接编码成gbk2312格式过程解析
网友评论