针对微型无人机检测难题,提出高效实时的新型检测框架。
UAV-DETR: DETR for Anti-Drone Target Detection

- 采用增强型小目标感知结构与滑动窗口自注意力机制
- 在自定义数据集上mAP50:95提升6.61%,参数减少39.8%
- 适合需要高精度实时反无人机检测的应用场景
无人机检测在众多安防与反无人机应用中至关重要。然而,现有基于深度学习的方法通常难以兼顾鲁棒的特征表示与计算效率,尤其在复杂背景和严重环境干扰下检测微型无人机时挑战更大。为此,我们提出UAV-DETR,一种融合小目标友好架构与实时检测能力的新框架。其核心包括:基于WTConv的增强主干网络与滑动窗口自注意力(SWSA-IFI)编码器,有效捕捉微小目标的高频结构细节,同时大幅降低参数开销;提出高效跨尺度特征重校准与融合网络(ECFRFN),抑制背景噪声并聚合多尺度语义信息;引入混合内CIoU与NWD损失策略,缓解标准IoU对小目标微小位置偏差的极端敏感性。大量实验表明,UAV-DETR在自建无人机数据集上相比基线RT-DETR提升6.61%(mAP50:95),参数量减少39.8%;在公开数据集DUT-ANTI-UAV上,精度提升1.4%,F1分数提升1.0%。结果证明,UAV-DETR在效率与精度间实现了更优平衡。代码已开源:https://github.com/wd-sir/UAVDETR。
原文摘要 · Abstract (English)
Drone detection is pivotal in numerous security and counter-UAV applications. However, existing deep learning-based methods typically struggle to balance robust feature representation with computational efficiency. This challenge is particularly acute when detecting miniature drones against complex backgrounds under severe environmental interference. To address these issues, we introduce UAV-DETR, a novel framework that integrates a small-target-friendly architecture with real-time detection capabilities. Specifically, UAV-DETR features a WTConv-enhanced backbone and a Sliding Window Self-Attention (SWSA-IFI) encoder, capturing the high-frequency structural details of tiny targets while drastically reducing parameter overhead. Furthermore, we propose an Efficient Cross-Scale Feature Recalibration and Fusion Network (ECFRFN) to suppress background noise and aggregate multi-scale semantics. To further enhance accuracy, UAV-DETR incorporates a hybrid Inner-CIoU and NWD loss strategy, mitigating the extreme sensitivity of standard IoU metrics to minor positional deviations in small objects. Extensive experiments demonstrate that UAV-DETR significantly outperforms the baseline RT-DETR on our custom UAV dataset (+6.61% in mAP50:95, with a 39.8% reduction in parameters) and the public DUT-ANTI-UAV benchmark (+1.4% in Precision, +1.0% in F1-Score). These results establish UAV-DETR as a superior trade-off between efficiency and precision in counter-UAV object detection. The code is available at https://github.com/wd-sir/UAVDETR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。