Keras读取保存的模型时, 产生错误[ValueError: Unknown activation function:relu6]

时间:2021-10-01 18:02:34

Solution:

from keras.utils.generic_utils import CustomObjectScope

with CustomObjectScope({'relu6': keras.applications.mobilenet.relu6,'DepthwiseConv2D': keras.applications.mobilenet.DepthwiseConv2D}):
    model = load_model('model_saved.hdf5')

官方github相关issue参考链接