DEYO: 使用YOLO的DETR|逐步训练的策略

文章预览

导读 我们设计了一种创新的训练方法，称为逐步训练（step-by-step training）。 具体来说，在训练的第一阶段，我们使用经典的检测器（采用一对多匹配策略预训练）来初始化端到端检测器的骨干网络和颈部网络。 在训练的第二阶段，我们将端到端检测器的骨干网络和颈部网络冻结，从而需要从头开始训练解码器。 摘要 DETRs 的训练范式严重依赖于在 ImageNet 数据集上预训练其骨干网络。然而，图像分类任务提供的监督信号有限以及一对一匹配策略导致 DETRs 的颈部网络预训练不足。此外，训练初期匹配的不稳定性造成了 DETRs 在优化目标方面的不一致。为了解决这些问题，我们设计了一种创新的训练方法，称为逐步训练（step-by-step training）。具体来说，在训练的第一阶段，我们使用经典的检测器（采用一对多匹配策略预训练）来初始化端到端检测器的 ………………………………