Point2RBox-v2: Rethinking Point-supervised Oriented Object Detection with Spatial Layout Among Instances
Yi Yu, Botao Ren, Peiyuan Zhang, Mingxin Liu, Junwei Luo, Shaofeng Zhang, Feipeng Da, Junchi Yan, Xue Yang
TL;DR
Point2RBox-v2 tackles point-supervised oriented object detection by exploiting the spatial layout among instances. It introduces a pair of layout-based losses—Gaussian overlap and Voronoi watershed—along with a consistency loss, edge guidance, and copy-paste augmentation to tightly constrain object size and orientation without heavy priors. Across diverse remote-sensing and retail datasets, the method achieves state-of-the-art results in end-to-end and pseudo-label settings, notably surpassing prior point-supervised approaches in dense scenes and approaching RBox-supervised performance in several benchmarks. The approach remains robust to annotation noise and does not require pre-trained priors, offering a practical, lightweight path toward accurate oriented detection from points, with clear limitations on sparse categories that lack layout cues.
Abstract
With the rapidly increasing demand for oriented object detection (OOD), recent research involving weakly-supervised detectors for learning OOD from point annotations has gained great attention. In this paper, we rethink this challenging task setting with the layout among instances and present Point2RBox-v2. At the core are three principles: 1) Gaussian overlap loss. It learns an upper bound for each instance by treating objects as 2D Gaussian distributions and minimizing their overlap. 2) Voronoi watershed loss. It learns a lower bound for each instance through watershed on Voronoi tessellation. 3) Consistency loss. It learns the size/rotation variation between two output sets with respect to an input image and its augmented view. Supplemented by a few devised techniques, e.g. edge loss and copy-paste, the detector is further enhanced. To our best knowledge, Point2RBox-v2 is the first approach to explore the spatial layout among instances for learning point-supervised OOD. Our solution is elegant and lightweight, yet it is expected to give a competitive performance especially in densely packed scenes: 62.61%/86.15%/34.71% on DOTA/HRSC/FAIR1M. Code is available at https://github.com/VisionXLab/point2rbox-v2.
