面向无人机航拍小目标检测的轻量级YOLOv8检测算法

来源：计算机工程与应用杂志2024年第17期北京时间：

作者：李岩超, 史卫亚, 冯灿

单位：1.河南工业大学 信息科学与工程学院，郑州 450001；2.河南工业大学 人工智能与大数据学院，郑州 450001

摘要：针对在无人机图像目标检测中复杂场景下目标特征难提取且小目标容易被淹没在噪声中的问题，提出一种基于YOLOv8s的无人机目标检测算法SC-YOLO。为了能够学习到感兴趣区域的位置细节，基于CA（coordinate attention）提出了SPM（self-position module）注意力。为了缓解Carafe上采样算子因为通道压缩所带来的影响，提出了CEM（Carafe enhancer module）。通过分析梯度增益函数与数据集中目标大小的关系，使WIoU_v3能够更加关注中、小目标的普通质量锚框，并且在VisDrone2019数据集上进行验证，得到WIoU_v3能够更加关注中、小目标的普通质量锚框的参数设置范围。改进后的YOLOv8s算法在VisDrone2019验证集上的平均均值精度（mAP）提高到43.1%，在测试集上的mAP提高到34.8%，在近几年同等规模的算法中拥有较好的检测性能；改进算法相较基准算法参数量仅增加1.1×106，浮点运算次数（FLOPs）增加1.5 GFLOPs，但在验证集以及测试集上检测精度分别提升了2.0和2.1个百分点；在Tinyperson数据集上的检测精度提高了1.4个百分点。

关键词：YOLOv8,Carafe,SGE注意力机制,坐标注意力机制,WIoU