该模型将Swin Transformer与YOLOv8s结合,引入CBAM和Soft-NMS,显著提升无人机图像中小目标和遮挡目标的检测性能。
Noor Ul Ain, Tahir · Zhe, Long · 张祖平 · 73047378 · El-Affendi, Mohammed Ahmed
Drones 2024
主干网络用 Swin Transformer 做全局特征提取,把远处小人小车也纳入上下文,减少小目标漏检。
CBAM 在通道和空间两个维度强化关键特征,Soft-NMS 管理重叠框,在遮挡场景下提升检测准确性。
相比 YOLOv8s 基线,在 VisDrone2019 数据集上 mAP 提升 4.8%,已经量化验证。