资源论文Bounding Box Regression with Uncertainty for Accurate Object Detection

Bounding Box Regression with Uncertainty for Accurate Object Detection

2019-09-09 | |  203 |   67 |   0

Abstract Large-scale object detection datasets (e.g., MS-COCO) try to defifine the ground truth bounding boxes as clear as possible. However, we observe that ambiguities are still introduced when labeling the bounding boxes. In this paper, we propose a novel bounding box regression loss for learning bounding box transformation and localization variance together. Our loss greatly improves the localization accuracies of various architectures with nearly no additional computation. The learned localization variance allows us to merge neighboring bounding boxes during non-maximum suppression (NMS), which further improves the localization performance. On MS-COCO, we boost the Average Precision (AP) of VGG-16 Faster R-CNN from 23.6% to 29.1%. More importantly, for ResNet-50-FPN Mask R-CNN, our method improves the AP and AP90 by 1.8% and 6.2% respectively, which signifificantly outperforms previous stateof-the-art bounding box refifinement methods. Our code and models are available at github.com/yihui-he/KL-Loss

上一篇:Attention Based Glaucoma Detection: A Large-scale Database and CNN Model

下一篇:Precise Detection in Densely Packed Scenes

用户评价
全部评价

热门资源

  • The Variational S...

    Unlike traditional images which do not offer in...

  • Learning to Predi...

    Much of model-based reinforcement learning invo...

  • Stratified Strate...

    In this paper we introduce Stratified Strategy ...

  • Learning to learn...

    The move from hand-designed features to learned...

  • A Mathematical Mo...

    Direct democracy, where each voter casts one vo...