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這個函數,在兩個版本中存在於不同的類中,所以不正確調用,就會報錯。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章