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

时间:2018-01-18 00:23:30

标签: python tensorflow machine-learning object-detection

我已经训练了一个自定义人物探测器,它不能很好地检测航拍镜头上的庇护,这是因为我的数据集缺乏一个人的航拍图像。我可以使用最新检查点继续训练并为我的模型添加另一个数据集(不同的tfrecord文件),还是必须使用更新的数据集重新开始训练?

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

1 个答案:

答案 0 :(得分:1)

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