Abstract:One-stage detector basically formulates object detection as dense classification and localization. The classification is usually optimized by Focal Loss and the box location is commonly learned under Dirac delta distribution. A recent trend for one-stage detectors is to introduce an individual prediction branch to estimate the quality of localization, where the predicted quality facilitates the classification to improve detection performance. This paper delves into the representations of the above three fundamental elements: quality estimation, classification and localization. Two problems are discovered in existing practices, including (1) the inconsistent usage of the quality estimation and classification between training and inference and (2) the inflexible Dirac delta distribution for localization when there is ambiguity and uncertainty in complex scenes. To address the problems, we design new representations for these elements. Specifically, we merge the quality estimation into the class prediction vector to form a joint representation of localization quality and classification, and use a vector to represent arbitrary distribution of box locations. The improved representations eliminate the inconsistency risk and accurately depict the flexible distribution in real data, but contain continuous labels, which is beyond the scope of Focal Loss. We then propose Generalized Focal Loss (GFL) that generalizes Focal Loss from its discrete form to the continuous version for successful optimization. On COCO test-dev, GFL achieves 45.0\% AP using ResNet-101 backbone, surpassing state-of-the-art SAPD (43.5\%) and ATSS (43.6\%) with higher or comparable inference speed, under the same backbone and training settings. Notably, our best model can achieve a single-model single-scale AP of 48.2\%, at 10 FPS on a single 2080Ti GPU. Code and models are available at this https URL.

Balanced One-Stage Object Detection by Enhancing the Effect of Positive Samples

Delving deep into the imbalance of positive proposals in two-stage object detection

Gradient Harmonized Single-stage Detector

PosNeg-Balanced Anchors with Aligned Features for Single-Shot Object Detection

Balance-Oriented Focal Loss with Linear Scheduling for Anchor Free Object Detection

AP-Loss for Accurate One-Stage Object Detection

Generalized Focal Loss: Learning Qualified and Distributed Bounding Boxes for Dense Object Detection

Integrating Guided Anchoring into One-stage Object Detectors

Semi-Anchored Detector for One-Stage Object Detection

Revisiting Foreground-Background Imbalance in Object Detectors

Auxiliary Detection Head for One-Stage Object Detection

Learning from Noisy Anchors for One-Stage Object Detection.

LapNet : Automatic Balanced Loss and Optimal Assignment for Real-Time Dense Object Detection

A dynamic label assignment strategy for one-stage detectors

DFL-Net: Effective Object Detection Via Distinguishable Feature Learning

Overlapping object detection with adaptive Gaussian sample division and asymmetric weighted loss

Robust one-stage object detection with location-aware classifiers

Improving Real-world Object Detection Using Balanced Loss

Single-Shot Balanced Detector for Geospatial Object Detection

Bridging the Gap Between Anchor-based and Anchor-free Detection via Adaptive Training Sample Selection

PETDet: Proposal Enhancement for Two-Stage Fine-Grained Object Detection