Progressive End-to-End Object Detection in Crowded Scenes

TL;DR

This paper introduces a progressive query-based detection framework that improves crowded scene object detection by refining predictions iteratively, significantly enhancing performance over existing methods.

Contribution

It proposes a novel progressive predicting method leveraging one-to-one label assignment to refine predictions, boosting query-based detector performance in crowded scenes.

Findings

Achieves 92.0% AP on CrowdHuman dataset

Outperforms box-based methods like MIP in crowded scenarios

Provides consistent improvements across various crowdedness levels

Abstract

In this paper, we propose a new query-based detection framework for crowd detection. Previous query-based detectors suffer from two drawbacks: first, multiple predictions will be inferred for a single object, typically in crowded scenes; second, the performance saturates as the depth of the decoding stage increases. Benefiting from the nature of the one-to-one label assignment rule, we propose a progressive predicting method to address the above issues. Specifically, we first select accepted queries prone to generate true positive predictions, then refine the rest noisy queries according to the previously accepted predictions. Experiments show that our method can significantly boost the performance of query-based detectors in crowded scenes. Equipped with our approach, Sparse RCNN achieves 92.0\% $\text{AP}$, 41.4\% $\text{MR}^{-2}$ and 83.2\% $\text{JI}$ on the challenging CrowdHuman \cite{shao2018crowdhuman} dataset, outperforming the box-based method MIP \cite{chu2020detection} that specifies in handling crowded scenarios. Moreover, the proposed method, robust to crowdedness, can still obtain consistent improvements on moderately and slightly crowded datasets like CityPersons \cite{zhang2017citypersons} and COCO \cite{lin2014microsoft}. Code will be made publicly available at https://github.com/megvii-model/Iter-E2EDET.