现代计算机2024,Vol.30Issue(22) :8-14.DOI:10.3969/j.issn.1007-1423.2024.22.002

基于改进YOLOv8的航拍图像目标检测算法

Aerial image target detection algorithm based on improved YOLOv8

程劲松 魏艳龙
现代计算机2024,Vol.30Issue(22) :8-14.DOI:10.3969/j.issn.1007-1423.2024.22.002

基于改进YOLOv8的航拍图像目标检测算法

Aerial image target detection algorithm based on improved YOLOv8

程劲松 1魏艳龙1
扫码查看

作者信息

  • 1. 太原师范学院计算机科学与技术学院,晋中 030600
  • 折叠

摘要

无人机航拍图像中的目标检测一直是研究的热点.与标准图像相比,图像背景具有小目标众多、目标尺度变化大等特点.因此,传统的目标检测算法不适合直接用于无人机图像中.针对这些问题,研究了一种基于YOLOv8的目标检测算法.首先,为提高多尺度目标检测精度,提出基于Large Selective Kernel Network(LSKNet)的C2F-L结构,通过动态调整网络的感受野,更有效地处理检测对象变化的上下文信息.引入Slim-neck结构,降低参数数量,提高模型检测效率.最后,使用WIoU损失函数提高网络模型的泛化能力和整体性能.在VisDrone2019数据集上实验表明,改进算法的mAP@0.5达到33.4%,比原始YOLOv8提高了1.1个百分点,计算量降低了7.4%.事实证明,改进的算法能有效提高航拍图像的目标检测精度.

Abstract

Object detection in drone aerial images has always been a hot topic of research.Compared with standard images,im-age backgrounds have many small objects and large object scale variations.Therefore,traditional object detection algorithms are not suitable for direct use in drone images.To address these issues,a YOLOv8-based object detection algorithm was studied.First,in order to improve the accuracy of multi-scale object detection,a C2F-L structure based on Large Selective Kernel Network(LSKNet)was proposed.By dynamically adjusting the receptive field of the network,the contextual information of the detected ob-ject changes can be more effectively processed.The Slim-neck structure was introduced to reduce the number of parameters and im-prove the model detection efficiency.Finally,the WIoU loss function was used to improve the generalization ability and overall per-formance of the network model.Experiments on the VisDrone2019 dataset show that the mAP@0.5 of the improved algorithm reaches 33.4%,which is 1.1 percentage point higher than the original YOLOv8 and the computational complexity is reduced by 7.4%.It has been proven that the improved algorithm can effectively improve the object detection accuracy of aerial images.

关键词

航拍图像/YOLOv8/目标检测/LSKNet/WIoU

Key words

aerial images/YOLOv8/object detection/LSKNet/WIoU

引用本文复制引用

出版年

2024
现代计算机
中大控股

现代计算机

影响因子:0.292
ISSN:1007-1423
段落导航相关论文