基于改进YOLOv8s的无人机航拍目标检测算法

针对无人机航拍图像存在目标小、目标遮挡等问题,提出一种改进YOLOv8s的目标检测算法YOLO-RC。在骨干网络结构中引入感受野空间注意力(RFA),避免卷积核参数共享,以提高模型的图像特征提取性能。改进C2f模块,引入深度分离卷积,减少模型的计算量。新增混合注意力卷积的小目标检测层,以改善对小目标检测精度。为充分考虑预测图像几何特征,使用MPDIoU损失函数优化网络。在无人机图像数据集VisDrone2019上的实验表明,所提改进算法的mAP@0.5为44.7%,较YOLOv8s提升了5.4个百分点,在新增小目标检测层的情况下,参数量降低了1.81×106。在DOTAv1.0数据集上,mAP@0.5提高了5.6个百分点。改进后的算法具有更强的鲁棒性,适用于无人机视角目标检测任务。