arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2404.04140cs.CVcs.LG

上下文感知的航空目标检测:利用目标间与背景关系

Context-Aware Aerial Object Detection: Leveraging Inter-Object and Background Relationships

Tsinghua University(清华大学);OpenGVLab(开放视觉实验室);Shanghai AI Laboratory(上海人工智能实验室)

Botao Ren, Botian Xu, Xue Yang, Yifan Pu, Jingyi Wang, Zhidong Deng

AI总结:

针对现代目标检测忽略目标间及目标-背景关系的问题,提出结合Transformer模型与CLIP特征的框架,在两阶段检测器基础上处理RoI候选框与CLIP Token并整合空间几何关系,在四个数据集上实现mAP₅₀提升,创最新结果。

摘要:

在大多数现代目标检测流程中,检测候选框是基于特征图独立处理的,因此忽略了目标与周围背景之间的潜在关系,而这些关系本可提供额外上下文以实现准确检测。由于航空影像几乎是正射影像,图像空间中的空间关系与物理世界中的空间关系高度一致,目标间及目标-背景关系变得尤为重要。为解决这一疏漏,我们提出一个框架,利用基于Transformer的模型和对比语言-图像预训练(CLIP)特征来捕捉此类关系。具体而言,基于两阶段检测器,我们将感兴趣区域(RoI)候选框视为token,辅以从多级图像段获得的CLIP Token。这些token随后通过Transformer编码器,其中特定空间和几何关系被整合到注意力权重中,这些权重会被自适应调制和正则化。此外,我们在CLIP Token上引入自监督约束以确保一致性。在三个基准数据集上的大量实验表明,我们的方法实现了持续改进,创下新的最先进结果:DOTA-v1.0上mAP₅₀提升1.37,DOTA-v1.5上提升5.30,DOTA-v2.0上提升2.30,DIOR-R上提升3.23。

英文摘要:

In most modern object detection pipelines, the detection proposals are processed independently given the feature map. Therefore, they overlook the underlying relationships between objects and the surrounding background, which could have provided additional context for accurate detection. Because aerial imagery is almost orthographic, the spatial relations in image space closely align with those in the physical world, and inter-object and object-background relationships become particularly significant. To address this oversight, we propose a framework that leverages the strengths of Transformer-based models and Contrastive Language-Image Pre-training (CLIP) features to capture such relationships. Specifically, Building on two-stage detectors, we treat Region of Interest (RoI) proposals as tokens, accompanied by CLIP Tokens obtained from multi-level image segments. These tokens are then passed through a Transformer encoder, where specific spatial and geometric relations are incorporated into the attention weights, which are adaptively modulated and regularized. Additionally, we introduce self-supervised constraints on CLIP Tokens to ensure consistency. Extensive experiments on three benchmark datasets demonstrate that our approach achieves consistent improvements, setting new state-of-the-art results with increases of 1.37 mAP$_{50}$ on DOTA-v1.0, 5.30 mAP$_{50}$ on DOTA-v1.5, 2.30 mAP$_{50}$ on DOTA-v2.0 and 3.23 mAP$_{50}$ on DIOR-R.

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

↑