arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.06503cs.CVcs.AI

改进航空图像中小型有向目标的检测

Improving the Detection of Small Oriented Objects in Aerial Images

Electrical and Electronics Engineering Institute(电气与电子工程研究所);University of the Philippines, Diliman(菲律宾大学迪利曼分校)

Chandler Timm C. Doloriel, Rhandley D. Cajote

AI总结:

本文针对航空图像中小型有向目标检测问题,提出 Attention-Points Network,通过 GALoss 和 BPLoss 增强分类与回归,并在 DOTA-v1.5 和 HRSC2016 上验证有效性。

摘要:

在大尺度航空图像中,仅占极小像素区域的小型有向目标由于其尺寸和方向而难以检测。现有的有向航空检测器已展现出有前景的结果,但主要关注方向建模,而对目标尺寸的关注较少。在本工作中,我们提出了一种通过增强有向目标检测模型的分类和回归任务来准确检测航空图像中小型有向目标的方法。我们设计了 Attention-Points Network,包含两个损失:Guided-Attention Loss(GALoss)和 Box-Points Loss(BPLoss)。GALoss 使用实例分割掩码作为真值来学习改进小目标检测所需的注意力特征。这些注意力特征随后被用于为 BPLoss 预测框点,BPLoss 确定这些点相对于目标有向边界框的位置。实验结果表明,我们的 Attention-Points Network 在包含小目标实例的标准有向航空数据集(DOTA-v1.5)和海事相关数据集(HRSC2016)上均有效。代码已公开。

英文摘要:

Small oriented objects that represent tiny pixel-area in large-scale aerial images are difficult to detect due to their size and orientation. Existing oriented aerial detectors have shown promising results but are mainly focused on orientation modeling with less regard to the size of the objects. In this work, we proposed a method to accurately detect small oriented objects in aerial images by enhancing the classification and regression tasks of the oriented object detection model. We designed the Attention-Points Network consisting of two losses: Guided-Attention Loss (GALoss) and Box-Points Loss (BPLoss). GALoss uses an instance segmentation mask as ground-truth to learn the attention features needed to improve the detection of small objects. These attention features are then used to predict box points for BPLoss, which determines the points' position relative to the target oriented bounding box. Experimental results show the effectiveness of our Attention-Points Network on a standard oriented aerial dataset with small object instances (DOTA-v1.5) and on a maritime-related dataset (HRSC2016). The code is publicly available.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑