【问题标题】:Is it possible to add another dataset on my trained model in tensorflow?是否可以在 tensorflow 中的训练模型上添加另一个数据集?
【发布时间】:2018-06-26 23:34:47
【问题描述】:
我已经训练了一个自定义人物检测器,它不能很好地检测航拍镜头上的人,这是因为我的数据集缺少人的航拍图像。我可以使用最新的检查点继续训练并为我的模型添加另一个数据集(不同的 tfrecord 文件),还是必须使用更新的数据集重新开始训练?
我重新训练了 Inception 模型来检测人,因为没有其他方法可以从预训练模型中移除其他 89 个对象。
【问题讨论】:
标签:
python
tensorflow
machine-learning
object-detection
【解决方案1】:
您绝对可以使用另一个数据集从检查点开始。然而,由于神经网络倾向于忘记他们已经学过的东西(这个问题被称为catastrophic forgetting),因此只对数据的一个子集进行训练可能不是一个好主意。最好创建一个包含新旧数据的新数据集,然后使用这些数据从检查点获取(类似于微调 Inception 模型的方式),或者重新开始训练过程。