计算机工程与应用2024，Vol.60Issue(24)：200-210,11.DOI:10.3778/j.issn.1002-8331.2405-0019
基于YOLOv8n的航拍图像小目标检测算法
Small Target Detection Algorithm for Aerial Images Based on YOLOv8n
摘要
Abstract
To address the issue of dense targets and mutual occlusion in small target detection for aerial images,this paper proposes a small target detection algorithm based on YOLOv8n for aerial images.The algorithm incorporates several key enhancements.Firstly,at the end of the backbone network,the Bottleneck is replaced in C2f with improved FasterNet,maintaining the number of channels while improving convergence speed.Secondly,the CBS activation function SiLU is replaced in SPPF with ReLU,setting the input negative value to zero,and then the SE attention mechanism is introduced to retain more small target features.Thirdly,the efficient multi-scale attention mechanism EMA is embeded in front of the detection head,obtaining more detailed information and enhancing small target attention.Finally,the baseline network loss function CIoU is replaced with Wise IoU,providing a gain allocation strategy that prioritizes common quality anchor frames and improving network generalization.Ablation and comparison experiments are conducted using the VisDrone2021 and RSOD datasets.Results show an increase in mAP@0.5 by 5.1 and 7.2 percentage points compared to baseline algo-rithms for each dataset.Additionally,mAP@0.5:0.95 improved by 4.4 and 2.1 percentage points,respectively.These findings demonstrate a notable enhancement in the accuracy of detection metrics.Generalization experiments on the publicly available dataset VOC2007+2012 show an improvement of 3.8 percentage points for mAP@0.5,demonstrating good robustness.关键词
航拍图像/小目标检测/YOLOv8n/FasterNet/SPPF模块/高效多尺度注意力机制(EMA)/Wise IoUKey words
aerial images/small object detection/YOLOv8n/FasterNet/spatial pyramid pooling fast(SPPF)/efficient multi-scale attention(EMA)/Wise IoU分类
信息技术与安全科学引用本文复制引用
齐向明,严萍萍,姜亮..基于YOLOv8n的航拍图像小目标检测算法[J].计算机工程与应用,2024,60(24):200-210,11.基金项目
国家自然科学基金面上项目(62173171). （62173171）
