オブジェクト検出は最近、パイプラインの最後の1つの微分不可能なコンポーネントであるNon-Maximum Suppression(NMS)を削除し、エンドツーエンドのシステムを構築するという画期的な成果を達成しました。ただし、1対1の予測を行う理由は十分に理解されていません。この論文では、最初に1対1の正のサンプル割り当てが重要な要素であるのに対し、以前の検出器での1対多の割り当ては推論において冗長な予測を引き起こすことを指摘します。第2に、驚くべきことに、1対1の割り当てを使用したトレーニングでも、以前の検出器は冗長な予測を生成することがわかりました。マッチングコストの分類コストが主な要素であると特定します。(1)以前の検出器はロケーションコストのみを考慮し、(2)分類コストを追加で導入することにより、以前の検出器は推論中に1対1の予測を即座に生成します。スコアギャップの概念を紹介して、マッチングコストの影響を調査します。分類コストは、トレーニングの反復で最高スコアのサンプルとして陽性サンプルを選択し、ロケーションコストのみによってもたらされるノイズの多い陽性サンプルを減らすことにより、スコアギャップを拡大します。最後に、混雑したシーンでのエンドツーエンドのオブジェクト検出の利点を示します。コードはhttps://github.com/PeizeSun/OneNetで入手できます。
Object detection has recently achieved a breakthrough for removing the last one non-differentiable component in the pipeline, Non-Maximum Suppression (NMS), and building up an end-to-end system. However, what makes for its one-to-one prediction has not been well understood. In this paper, we first point out that one-to-one positive sample assignment is the key factor, while, one-to-many assignment in previous detectors causes redundant predictions in inference. Second, we surprisingly find that even training with one-to-one assignment, previous detectors still produce redundant predictions. We identify that classification cost in matching cost is the main ingredient: (1) previous detectors only consider location cost, (2) by additionally introducing classification cost, previous detectors immediately produce one-to-one prediction during inference. We introduce the concept of score gap to explore the effect of matching cost. Classification cost enlarges the score gap by choosing positive samples as those of highest score in the training iteration and reducing noisy positive samples brought by only location cost. Finally, we demonstrate the advantages of end-to-end object detection on crowded scenes. The code is available at: https://github.com/PeizeSun/OneNet.