在keras中连接两个以上的cnn模型

Question

我在 CNN 中使用 keras 制作模型，例如 MNIST.h5,cifar10.h5,dogs_and_cats_classification.h5。但是这些模型仅限于自己的类，例如 mnist 仅预测手写数字或 dogs_and_cats_classification.h5分别预测狗或猫的输出。 但我想制作一个单一模型（all.h5），以便该模型可以预测数字、狗、猫或其他一些。 那么他们有什么办法在 keras 中做这样的事情吗？

请给我建议.. 谢谢。

Answer 1

您可以删除最后一个密集层，添加一个新的密集层，然后重新训练您的模型，或者您可以创建一个多输出集成，您可以从 keras 常见问题解答中了解更多信息。