首页|改进YOLOv8的无人机空中图像目标检测算法

改进YOLOv8的无人机空中图像目标检测算法

Improved YOLOv8-Based Target Detection Algorithm for UAV Aerial Image

扫码查看
针对无人机航拍图像小目标的检测精度低且易被漏检和误检的挑战,本研究提出了一种基于YOLOv8的改进无人机图像目标检测算法.首先,采用CoordAtt注意机制增强了骨干网络的特征提取能力,从而减少了来自背景的干扰.然后,使用增加小目标检测层的BiFPN特征融合网络,提升模型对小目标的感知能力.此外,还设计并提出了一个多层融合模块,以有效地集成浅层和深层信息.使用增强的MPDIoU损失函数进一步提高了检测性能.基于公开的VisDrone2019数据集的实验结果表明,改进后的模型相比于YOLOv8基线模型,mAP@0.5提高了20%,改进方法提高了模型对小目标的检测精度.
In response to the challenge of low detection accuracy and susceptibility to missed and false detections of small targets in unmanned aerial vehicles(UAVs)aerial images,an improved UAV image target detection algorithm based on YOLOv8 was proposed in this study.To begin with,the CoordAtt attention mechanism was employed to enhance the feature extraction capability of the backbone network,thereby reducing interference from backgrounds.Additionally,the BiFPN feature fusion network with an added small object detection layer was used to enhance the model's ability to perceive for small objects.Furthermore,a multi-level fusion module was designed and proposed to effectively integrate shallow and deep information.The use of an enhanced MPDIoU loss function further improved detection performance.The experimental results based on the publicly available VisDrone2019 dataset showed that the improved model outperformed the YOLOv8 baseline model,mAP@0.5 improved by 20%,and the improved method improved the detection accuracy of the model for small targets.

UAVYOLOv8Attentional mechanismsMulti-scale detectionMPDIoU

姜贸翔、司占军

展开 >

天津科技大学 人工智能学院,天津 300457

无人机 YOLOv8 注意力机制 多尺度目标检测 MPDIoU

2024

数字印刷
中国印刷科学技术研究所

数字印刷

北大核心
ISSN:2095-9540
年,卷(期):2024.(4)
  • 2