Keras读取模型时报错ValueError: Unknown activation function:relu6

这个是一个Keras不同版本的问题,function改变了,网上一搜,排头的全是解决老版本的,现在用的是Keras版本是:2.2.2
以下是收集到的两个Keras版本解决方法:

  • Keras 2.1:
    model = load_model(model_path, custom_objects={‘relu6’: keras.applications.mobilenet.relu6,‘DepthwiseConv2D’: keras.applications.mobilenet.DepthwiseConv2D})
  • Keras 2.2:
    model = load_model(model_path, custom_objects={‘relu6’: keras.layers.ReLU(6.), ‘DepthwiseConv2D’: keras.layers.DepthwiseConv2D})

对比可以发现,relu6这个函数,在两个版本中存在于不同的类中,所以不正确调用,就会报错。

你可能感兴趣的:(Keras)