在Keras中,如果存在自定义layer或者loss,需要在load_model()中以字典形式指定layer或loss。
model = load_model(‘My_model.h5’, custom_objects={‘NEW_loss’:NEW_loss,‘NEW_LAYER’: NEW_LAYER})
在Keras中,如果存在自定义layer或者loss,需要在load_model()中以字典形式指定layer或loss。
model = load_model(‘My_model.h5’, custom_objects={‘NEW_loss’:NEW_loss,‘NEW_LAYER’: NEW_LAYER})