我最初用 5 个类训练了我的深度学习模型,现在我想添加另一个类,而不需要为这 5 个类重新训练整个模型。我怎样才能做到这一点?
如何为现有的训练有素的深度学习模型添加新类别?
数据挖掘
深度学习
喀拉斯
美国有线电视新闻网
计算机视觉
迁移学习
2022-02-22 11:02:00
2个回答
如果不重新训练模型的至少一部分,你就无法做到这一点。
您必须将现有的输出层替换为具有所需神经元数量的新输出层。当然,这意味着您必须至少重新训练最后一层或最后几层。但是您可以冻结所有其他层的权重。
trainable在 keras 中,您可以通过将其属性设置为 来冻结特定层的权重False。
如何在没有对其他类进行完整训练的情况下向训练模型添加更多类:
- 迁移学习两次
- 持续学习方法
- 正则化
- 扩张
- 排演
其它你可能感兴趣的问题