Overlap Sampler for Region-Based Object Detection

Joya Chen, Bin Luo, Qi Wu, Jia Chen, Xuezheng Peng; The IEEE Winter Conference on Applications of Computer Vision (WACV), 2020, pp. 767-775

Abstract


The top accuracy of object detection to date is led by region-based approaches, where the per-region stage is responsible for recognizing proposals generated by the region proposal network. In that stage, sampling heuristics (e.g., OHEM, IoU-balanced sampling) is always applied to select a part of examples during training. But nowadays, existing samplers ignore the overlaps among examples, which may result in some low-quality predictions preserved. To mitigate the issue, we propose Overlap Sampler that selects examples according to the overlaps among examples, which enables the training to focus on the important examples. Benefitted from it, the Faster R-CNN could obtain impressively 1.5 points higher Average Precision (AP) on the challenging COCO benchmark, a state-of-the-art result among existing samplers for region-based detectors. Moreover, the proposed sampler also yields considerable improvements for the instance segmentation task. Our code is released at https://github.com/ChenJoya/overlap-sampler.

Related Material


[pdf]
[bibtex]
@InProceedings{Chen_2020_WACV,
author = {Chen, Joya and Luo, Bin and Wu, Qi and Chen, Jia and Peng, Xuezheng},
title = {Overlap Sampler for Region-Based Object Detection},
booktitle = {The IEEE Winter Conference on Applications of Computer Vision (WACV)},
month = {March},
year = {2020}
}