【问题标题】: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 模型的方式),或者重新开始训练过程。

    【讨论】:

      猜你喜欢
      • 2018-03-28
      • 1970-01-01
      • 1970-01-01
      • 2018-10-23
      • 1970-01-01
      • 2015-10-25
      • 2020-10-22
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多