README.md 3.7 KB

CrowdDet

Detection in Crowded Scenes: One Proposal, Multiple Predictions

Abstract

We propose a simple yet effective proposal-based object detector, aiming at detecting highly-overlapped instances in crowded scenes. The key of our approach is to let each proposal predict a set of correlated instances rather than a single one in previous proposal-based frameworks. Equipped with new techniques such as EMD Loss and Set NMS, our detector can effectively handle the difficulty of detecting highly overlapped objects. On a FPN-Res50 baseline, our detector can obtain 4.9% AP gains on challenging CrowdHuman dataset and 1.0% MR^−2 improvements on CityPersons dataset, without bells and whistles. Moreover, on less crowed datasets like COCO, our approach can still achieve moderate improvement, suggesting the proposed method is robust to crowdedness. Code and pre-trained models will be released at https://github.com/megvii-model/CrowdDetection.

Results and Models

Backbone RM Style Mem (GB) Inf time (fps) box AP Config Download
R-50-FPN False pytorch 4.4 - 90.0 config model | log
R-50-FPN True pytorch 4.8 - 90.32 config model | log

Note:

  • RM indicates whether to use the refine module.
  • The dataset for training and testing this model is CrowdHuman, and the metric of box AP is calculated by mmdet/evaluation/metrics/crowdhuman_metric.py.

Citation

@inproceedings{Chu_2020_CVPR,
  title={Detection in Crowded Scenes: One Proposal, Multiple Predictions},
  author={Chu, Xuangeng and Zheng, Anlin and Zhang, Xiangyu and Sun, Jian},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
  month = {June},
  year = {2020}
}