分类：函数模型

2020
10-07

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

当我们导入的模型含有自定义层或者自定义函数时，需要使用custom_objects来指定目标层或目标函数。例如：我的一个模型含有自定义层“SincConv1D”，需要使用下面的代码导入：fromkeras.modelsimportload_modelmodel=load_model('model.h5',custom_objects={'SincConv1D':SincConv1D})如果不加custom_objects指定目标层Layer，则会出现以下报错：ValueError:Unknownlayer:SincConv1D同样的，当我的模型含有自定义函数“my... 继续阅读 >