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