ObjectBox: From Centers to Boxes for Anchor-Free Object Detection
This addresses the issue of dataset-dependent tuning in object detection for computer vision applications, though it appears incremental as it builds on existing anchor-free approaches.
The paper tackles the problem of scale bias in object detection by proposing ObjectBox, an anchor-free method that uses only object centers as positive samples and treats all objects equally across feature levels, achieving favorable performance compared to state-of-the-art methods on MS-COCO and PASCAL VOC datasets.
We present ObjectBox, a novel single-stage anchor-free and highly generalizable object detection approach. As opposed to both existing anchor-based and anchor-free detectors, which are more biased toward specific object scales in their label assignments, we use only object center locations as positive samples and treat all objects equally in different feature levels regardless of the objects' sizes or shapes. Specifically, our label assignment strategy considers the object center locations as shape- and size-agnostic anchors in an anchor-free fashion, and allows learning to occur at all scales for every object. To support this, we define new regression targets as the distances from two corners of the center cell location to the four sides of the bounding box. Moreover, to handle scale-variant objects, we propose a tailored IoU loss to deal with boxes with different sizes. As a result, our proposed object detector does not need any dataset-dependent hyperparameters to be tuned across datasets. We evaluate our method on MS-COCO 2017 and PASCAL VOC 2012 datasets, and compare our results to state-of-the-art methods. We observe that ObjectBox performs favorably in comparison to prior works. Furthermore, we perform rigorous ablation experiments to evaluate different components of our method. Our code is available at: https://github.com/MohsenZand/ObjectBox.