基于改进YOLOv8s的无人机目标检测算法
DOI:
作者:
作者单位:

长安大学信息工程学院 西安 710000

作者简介:

通讯作者:

中图分类号:

TN29

基金项目:


UAV target detection algorithm based on improved YOLOv8s
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对目前无人机航拍图像目标尺寸较小,图像背景复杂,导致现有的无人机目标检测算法检测精度较低的问题,本文提出一种改进YOLOv8s的无人机目标检测算法。首先使用可变形卷积替换标准卷积,以增强网络对不规则形状目标的特征提取能力;然后使用可分离大核注意力机制LSKA改进SPPF模块,改善因目标尺度差异较大导致检测精度较低的问题。在网络颈部结合双向特征金字塔网络Bi-FPN实现多尺度特征融合,改善网络对小目标的漏检和错检问题。在网络头部,使用自注意力机制动态检测头DyHead替换原检测头,增强对遮挡物体和小目标的检测能力。最后,针对数据集中存在大量低质量样本对训练过程产生负面影响的问题,使用Wise-IOU损失函数,提升模型收敛速度和检测精度。实验结果表明,改进后的方法在VisDrone2019数据集上获得了41.7%的mAP,与原YOLOv8s算法相比,mAP50提升了3.0%,mAP50:95提升了1.9%,参数量下降了17.5%,计算量下降了12.63%。实现了模型轻量化和检测精度双重提升。

    Abstract:

    Aiming at the problem that the target size of the current UAV aerial images is small and the image background is complex, which leads to low detection accuracy of existing UAV target detection algorithms, this article proposes an improved YOLOv8s UAV target detection algorithm. First, deformable convolution is used to replace standard convolution to enhance the network's feature extraction ability for irregularly shaped targets; Then the separable large-core attention mechanism LSKA is used to improve the SPPF module to improve the problem of low detection accuracy due to large differences in target scales. The weighted bi-directional feature pyramid network Bi-FPN is combined at the neck of the network to achieve multi-scale feature fusion and improve the network's missed detection and false detection of small targets. At the head of the network, the dynamic detection head DyHead is used to replace the original detection head to enhance the detection ability of occluded objects and small targets. Finally, in order to solve the problem that a large number of low-quality samples in the dataset have a negative impact on the training process, the Wise-IOU loss function was used to improve the model convergence speed and detection accuracy. Experimental results show that the improved method achieved 41.7% mAP on the VisDrone2019 dataset. Compared with the original YOLOv8s algorithm, mAP50 increased by 3.0%, mAP50:95 increased by 1.9%, the number of parameters decreased by 17.5%, and the amount of calculation dropped by 12.63%. It achieves both model lightweight and detection accuracy.

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-03-27
  • 最后修改日期:2024-05-31
  • 录用日期:2024-06-03
  • 在线发布日期:
  • 出版日期: