ValueError: Unknown layer: InstanceNormalization

使用keras的load_model出现如上错误

原因是我们在训练模型的时候使用了InstanceNormalization这个归一化层代替了之前的BatchNormalization这个归一化层。而在我们使用keras的load_model这个函数加载保存的模型时，它不知道这个InstanceNormalization是什么东西。所以它会报错。
那我们就要让它知道这是什么意思，所以在调用load_model这个函数时加一点东西就可以了：

model1 = load_model('./saved_model/model1.hdf5',custom_objects={'InstanceNormalization': InstanceNormalization})

当然在这个之前我们需要在前面导入一下，代码如下：

from keras_contrib.layers.normalization.instancenormalization import InstanceNormalization

如果没有keras_contrib的话。那需要安装一下：

pip install git+https://www.github.com/keras-team/keras-contrib.git