arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.19696cs.CV

用于航拍图像小目标检测的跨层特征金字塔Transformer

Cross-Layer Feature Pyramid Transformer for Small Object Detection in Aerial Images

National Key Lab of Autonomous Intelligent Unmanned Systems, School of Automation, Beijing Institute of Technology(国家级自主智能无人机系统实验室,自动化学院,北京理工大学)

Zewen Du, Zhenjiang Hu, Guiyu Zhao, Ying Jin, Hongbin Ma

AI总结:

本文提出跨层特征金字塔Transformer(CFPT),通过CCA和CSA模块实现单步高效跨层交互,结合全局上下文与相对位置编码,在航拍图像小目标检测中以更低计算成本超越现有特征金字塔网络。

摘要:

由于目标通常尺寸较小,航拍图像中的目标检测一直是一项具有挑战性的任务。大多数当前的检测器优先开发新的检测框架,往往忽略了对特征金字塔网络等基础组件的研究。在本文中,我们引入了Cross-Layer Feature Pyramid Transformer (CFPT),这是一种专门为航拍图像小目标检测设计的无上采样器特征金字塔网络。CFPT结合了两个精心设计的具有线性计算复杂度的注意力模块:Cross-Layer Channel-Wise Attention (CCA)和Cross-Layer Spatial-Wise Attention (CSA)。CCA通过划分通道维度的token组来感知沿空间维度的跨层全局信息,从而实现跨层交互;而CSA通过划分空间维度的token组来感知沿通道维度的跨层全局信息,从而实现跨层交互。通过整合这些模块,CFPT能够在单步内实现高效的跨层交互,从而避免了与逐元素求和和逐层传输相关的语义鸿沟和信息损失。此外,CFPT结合了全局上下文信息,提高了小目标的检测性能。为了进一步增强跨层交互过程中的位置感知能力,我们提出了基于层间相互感受野的Cross-Layer Consistent Relative Positional Encoding (CCPE)。我们在三个具有挑战性的航拍图像目标检测数据集上评估了CFPT的有效性:VisDrone2019-DET、TinyPerson和xView。大量实验表明,CFPT在取得更低计算成本的同时,性能优于当前最先进的特征金字塔网络。代码可在https://github.com/duzw9311/CFPT获取。

英文摘要:

Object detection in aerial images has always been a challenging task due to the generally small size of the objects. Most current detectors prioritize the development of new detection frameworks, often overlooking research on fundamental components such as feature pyramid networks. In this paper, we introduce the Cross-Layer Feature Pyramid Transformer (CFPT), a novel upsampler-free feature pyramid network designed specifically for small object detection in aerial images. CFPT incorporates two meticulously designed attention blocks with linear computational complexity: Cross-Layer Channel-Wise Attention (CCA) and Cross-Layer Spatial-Wise Attention (CSA). CCA achieves cross-layer interaction by dividing channel-wise token groups to perceive cross-layer global information along the spatial dimension, while CSA enables cross-layer interaction by dividing spatial-wise token groups to perceive cross-layer global information along the channel dimension. By integrating these modules, CFPT enables efficient cross-layer interaction in a single step, thereby avoiding the semantic gap and information loss associated with element-wise summation and layer-by-layer transmission. In addition, CFPT incorporates global contextual information, which improves detection performance for small objects. To further enhance location awareness during cross-layer interaction, we propose the Cross-Layer Consistent Relative Positional Encoding (CCPE) based on inter-layer mutual receptive fields. We evaluate the effectiveness of CFPT on three challenging object detection datasets in aerial images: VisDrone2019-DET, TinyPerson, and xView. Extensive experiments demonstrate that CFPT outperforms state-of-the-art feature pyramid networks while incurring lower computational costs. The code is available at https://github.com/duzw9311/CFPT.

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

↑