深層学習ベースの検出器は通常、同じオブジェクトの多数の重複検出を含む、オブジェクト境界ボックスの冗長セットを生成します。次に、これらのボックスは、対象のオブジェクトごとに正確に1つの境界ボックスを選択するために、非最大抑制(NMS)を使用してフィルタリングされます。この貪欲なスキームは単純で、孤立したオブジェクトに十分な精度を提供しますが、異なるオブジェクトのボックスを保持し、重複検出を抑制する必要があるため、混雑した環境では失敗することがよくあります。この作業では、オブジェクトの新しいサブセットが各反復で検出される代替の反復スキームを開発します。前の反復で検出されたボックスは、同じオブジェクトが2回検出されないようにするために、次の反復でネットワークに渡されます。この反復スキームは、トレーニングと推論の手順をわずかに変更するだけで、1ステージと2ステージの両方のオブジェクト検出器に適用できます。 4つのデータセットで2つの異なるベースライン検出器を使用して広範な実験を実行し、ベースラインを大幅に上回って、CrowdHumanおよびWiderPersonデータセットで最先端のパフォーマンスを実現しました。ソースコードとトレーニング済みモデルは、https://github.com/saic-vul/iterdetで入手できます。
Deep learning-based detectors usually produce a redundant set of object bounding boxes including many duplicate detections of the same object. These boxes are then filtered using non-maximum suppression (NMS) in order to select exactly one bounding box per object of interest. This greedy scheme is simple and provides sufficient accuracy for isolated objects but often fails in crowded environments, since one needs to both preserve boxes for different objects and suppress duplicate detections. In this work we develop an alternative iterative scheme, where a new subset of objects is detected at each iteration. Detected boxes from the previous iterations are passed to the network at the following iterations to ensure that the same object would not be detected twice. This iterative scheme can be applied to both one-stage and two-stage object detectors with just minor modifications of the training and inference procedures. We perform extensive experiments with two different baseline detectors on four datasets and show significant improvement over the baseline, leading to state-of-the-art performance on CrowdHuman and WiderPerson datasets. The source code and the trained models are available at https://github.com/saic-vul/iterdet.