This paper addresses the challenges in drone object detection—complex backgrounds and numerous small objects—aiming to improve detection accuracy without increasing parameter count. By analyzing YOLOv10n’s architecture, we propose SSA-YOLO, a lightweight enhanced version with two key improvements: the SSA (Small Shuffle Attention) module in the backbone strengthens small-object feature extraction and fusion via spatial attention, while the USKConv (UltraLight Selective Kernel Convolution) module in the Neck optimizes target localization and classification through dynamic feature processing. Experiments on Visdrone2019 show SSA-YOLO improves mAP50 by 2.2% over YOLOv10n, with only a 7.1% parameter increase, and maintains generalization across datasets, validating the proposed strategies. This work supports practical drone detection applications and offers insights for optimizing small-object models on mobile devices.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

SSA-YOLO: A Lightweight Drone Imagery Detection Algorithm Improving YOLOv10

  • Che Guowei,
  • Zhang Mengqi,
  • Zhao Ruijun,
  • Wang Chengyao

摘要

This paper addresses the challenges in drone object detection—complex backgrounds and numerous small objects—aiming to improve detection accuracy without increasing parameter count. By analyzing YOLOv10n’s architecture, we propose SSA-YOLO, a lightweight enhanced version with two key improvements: the SSA (Small Shuffle Attention) module in the backbone strengthens small-object feature extraction and fusion via spatial attention, while the USKConv (UltraLight Selective Kernel Convolution) module in the Neck optimizes target localization and classification through dynamic feature processing. Experiments on Visdrone2019 show SSA-YOLO improves mAP50 by 2.2% over YOLOv10n, with only a 7.1% parameter increase, and maintains generalization across datasets, validating the proposed strategies. This work supports practical drone detection applications and offers insights for optimizing small-object models on mobile devices.