用少量水平框或点标注实现高效旋转目标检测,降低人工成本。
Partial Weakly-Supervised Oriented Object Detection
- 基于部分弱标注(水平框或点)设计新框架,可高效利用大量无标签数据。
- 在DOTA和DIOR数据集上性能接近甚至超过传统半监督方法。
- 适合标注资源有限但需高精度旋转检测的应用场景。
面向各类场景中日益增长的旋转目标检测需求,现有方法面临标注成本高的挑战。当前主流算法分为三类:完全监督(完整旋转框标注)、半监督(部分旋转框)和弱监督(水平框或点标注),但均难以平衡性能与标注效率。为此,本文提出首个部分弱监督旋转目标检测(PWOOD)框架,仅依赖水平框或单点等弱标注,显著降低标注成本;设计了面向方向与尺度感知的学生模型(OS-Student),能从少量无关方向或尺度的弱标注中学习关键特征;提出类别无关伪标签过滤策略(CPF),有效缓解模型对固定阈值的敏感性。在DOTA-v1.0/v1.5/v2.0和DIOR数据集上的实验证明,PWOOD性能媲美甚至超越传统半监督方法。代码将公开。
原文摘要 · Abstract (English)
The growing demand for oriented object detection (OOD) across various domains has driven significant research in this area. However, the high cost of dataset annotation remains a major concern. Current mainstream OOD algorithms can be mainly categorized into three types: (1) fully supervised methods using complete oriented bounding box (OBB) annotations, (2) semi-supervised methods using partial OBB annotations, and (3) weakly supervised methods using weak annotations such as horizontal boxes or points. However, these algorithms inevitably increase the cost of models in terms of annotation speed or annotation cost. To address this issue, we propose: (1) the first Partial Weakly-Supervised Oriented Object Detection (PWOOD) framework based on partially weak annotations (horizontal boxes or single points), which can efficiently leverage large amounts of unlabeled data, significantly outperforming weakly supervised algorithms trained with partially weak annotations, also offers a lower cost solution; (2) Orientation-and-Scale-aware Student (OS-Student) model capable of learning orientation and scale information with only a small amount of orientation-agnostic or scale-agnostic weak annotations; and (3) Class-Agnostic Pseudo-Label Filtering strategy (CPF) to reduce the model's sensitivity to static filtering thresholds. Comprehensive experiments on DOTA-v1.0/v1.5/v2.0 and DIOR datasets demonstrate that our PWOOD framework performs comparably to, or even surpasses traditional semi-supervised algorithms. Our code will be made publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。