我定义了一个名为CapsuleLayer的自定义层。实际模型已在单独的类中定义。我已将权重加载到实际模型中，并将模型保存在.h5文件中。但是，当我尝试使用load_model(filepath)加载模型时，出现错误
加载保存的模型时，如何将自定义图层合并到模型中。
最佳答案
C.f. Keras常见问题解答，"Handling custom layers (or other custom objects) in saved models":
就您而言，model = load_model('my_model.h5', custom_objects={'CapsuleLayer': CapsuleLayer})应该可以解决您的问题。
关于python - ValueError : Unknown layer: CapsuleLayer，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/50837728/