python - 值错误 : Unknown layer: CapsuleLayer

Question

我已经定义了一个名为 CapsuleLayer 的自定义层。实际模型已在单独的类中定义。我已将权重加载到实际模型中，并将模型保存在 .h5 文件中。但是，当我尝试使用 load_model(filepath) 加载模型时，出现错误

ValueError: Unknown layer: CapsuleLayer

如何在加载保存的模型时将自定义层合并到我的模型中。

Answer 1

比照Keras 常见问题解答，"Handling custom layers (or other custom objects) in saved models" :

If the model you want to load includes custom layers or other custom classes or functions, you can pass them to the loading mechanism via the custom_objects argument:

from keras.models import load_model
# Assuming your model includes instance of an "AttentionLayer" class
model = load_model('my_model.h5', custom_objects={'AttentionLayer': AttentionLayer})

Alternatively, you can use a custom object scope:

from keras.utils import CustomObjectScope

with CustomObjectScope({'AttentionLayer': AttentionLayer}):
    model = load_model('my_model.h5')

Custom objects handling works the same way for load_model, model_from_json, model_from_yaml:

from keras.models import model_from_json
model = model_from_json(json_string, custom_objects={'AttentionLayer': AttentionLayer})

---

在您的情况下，model = load_model('my_model.h5', custom_objects={'CapsuleLayer': CapsuleLayer}) 应该可以解决您的问题。