python - 是否可以在我的 tensorflow 训练模型上添加另一个数据集？

Question

我已经训练了一个自定义的人物检测器，它在检测航拍镜头上的人时效果不佳，这是因为我的数据集缺乏人的航拍图像。我可以使用最新的检查点继续训练并为我的模型添加另一个数据集(不同的 tfrecord 文件)，还是必须使用更新的数据集重新启动训练？

我重新训练了 Inception 模型以仅检测人员，因为没有其他方法可以从预训练模型中删除其他 89 个对象。

Answer 1

您绝对可以使用另一个数据集从检查点开始。然而，仅对数据的子集进行训练可能不是一个好主意，因为神经网络倾向于忘记它们已经学到的东西(这个问题称为 catastrophic forgetting )。创建一个包含旧数据和新数据的新数据集可能是一个更好的主意，然后使用这些数据从检查点获取(类似于微调 Inception 模型的方式)，或者重新开始训练过程。