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

我已经训练了一个自定义的人员检测器,但在检测空中拍摄的人员时效果不佳,这是因为我的数据集中缺乏人物的空中图像。我能否继续使用最新的检查点进行训练,并为我的模型添加另一个数据集(不同的tfrecord文件),还是必须使用更新后的数据集重新开始训练?

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


回答:

你当然可以使用另一个数据集从检查点开始训练。然而,仅使用数据的一个子集进行训练可能不是一个好主意,因为神经网络有忘记已学内容的倾向(这个问题被称为灾难性遗忘)。更好的做法是创建一个包含旧数据和新数据的新数据集,然后使用这些数据从检查点继续(类似于你对Inception模型进行微调的方式),或者重新开始训练过程。

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注