arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.11472cs.CV

Delving into the Imbalance of Positive Proposals in Two-stage Object Detection

Waseda University(早稻田大学);Tencent AI Lab(腾讯人工智能实验室);Nanjing University of Science and Technology(南京理工大学)

Zheng Ge, Zequn Jie, Xin Huang, Chengzheng Li, Osamu Yoshie

英文摘要:

Imbalance issue is a major yet unsolved bottleneck for the current object detection models. In this work, we observe two crucial yet never discussed imbalance issues. The first imbalance lies in the large number of low-quality RPN proposals, which makes the R-CNN module (i.e., post-classification layers) become highly biased towards the negative proposals in the early training stage. The second imbalance stems from the unbalanced ground-truth numbers across different testing images, resulting in the imbalance of the number of potentially existing positive proposals in testing phase. To tackle these two imbalance issues, we incorporates two innovations into Faster R-CNN: 1) an R-CNN Gradient Annealing (RGA) strategy to enhance the impact of positive proposals in the early training stage. 2) a set of Parallel R-CNN Modules (PRM) with different positive/negative sampling ratios during training on one same backbone. Our RGA and PRM can totally bring 2.0% improvements on AP on COCO minival. Experiments on CrowdHuman further validates the effectiveness of our innovations across various kinds of object detection tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑