无人机检测模型

该模型将Swin Transformer与YOLOv8s结合,引入CBAM和Soft-NMS,显著提升无人机图像中小目标和遮挡目标的检测性能。

Noor Ul Ain, Tahir · Zhe, Long · 张祖平 · 73047378 · El-Affendi, Mohammed Ahmed

Drones 2024

参数信息

平均检测精度提升
4.8 %

技术优势

小目标不漏检

主干网络用 Swin Transformer 做全局特征提取,把远处小人小车也纳入上下文,减少小目标漏检。

遮挡也能检准

CBAM 在通道和空间两个维度强化关键特征,Soft-NMS 管理重叠框,在遮挡场景下提升检测准确性。

精度实打实提升

相比 YOLOv8s 基线,在 VisDrone2019 数据集上 mAP 提升 4.8%,已经量化验证。

应用场景