改进YOLOv5s的轻量化无人机航拍小目标识别算法
DOI:
CSTR:
作者:
作者单位:

中北大学信息与通信工程学院 太原 030051

作者简介:

通讯作者:

中图分类号:

TN911.73

基金项目:

山西省基础研究计划青年科学研究项目(202303021212195)、中国博士后科学基金第76批面上项目(2024M76303)、四川省机器人与智能系统国际联合研究中心开放研究课题重点项目(JQZN2023-002)资助


Improve the small target recognition algorithm of YOLOv5s for lightweight UAV aerial photography
Author:
Affiliation:

College of Information and Communications Engineering,North University of China,Taiyuan 030051,China

Fund Project:

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

    无人机航拍是目前主流的物体检测技术之一,该任务中面临目标物体小、尺度变化大、复杂背景干扰等问题。如何在有限的计算资源下提升检测精度是一个重要的挑战。针对以上问题,提出了一种轻量级无人机航拍目标检测方法。首先设计了一种层次依赖感知剪枝算法减少模型的冗余计算。此外,将检测头分辨率提升至160×160以增强小目标检测能力,利用GhostConv替换网络的标准卷积块以减少计算冗余,并引入紧凑型架构StarNet重新设计Neck网络中的C3模块以减少特征融合过程的复杂度,增强特征表达能力。最后在Backbone层引入注意力机制来提高模型的特征提取能力。实验结果表明:在VisDrone2019数据集中,模型的mAP_0.5提升了1.8%。同时相比于原模型,参数量下降了50.4%,计算量降低了35.44%。综上所述,模型满足无人机平台在小目标检测任务中对精度和轻量化的需求。

    Abstract:

    UAV aerial photography is one of the mainstream object detection technologies, and this task faces problems such as small target objects, large scale changes, and complex background interference. How to improve the detection accuracy with limited computing resources is an important challenge. In order to solve the above problems, a lightweight UAV aerial target detection method was proposed. Firstly, a hierarchical dependence-aware pruning algorithm was designed to reduce the redundancy of the model. In addition, the resolution of the detection head is increased to 160×160 to enhance the detection ability of small targets, the standard convolution blocks of the network are replaced by GhostConv to reduce the computational redundancy, and the C3 module in the Neck network is redesigned by introducing the compact architecture StarNet to reduce the complexity of the feature fusion process and enhance the feature expression ability. Finally, the attention mechanism is introduced in the backbone layer to improve the feature extraction ability of the model. The experimental results show that in the VisDrone2019 dataset, the mAP_0.5 of the model is increased by 1.8%. At the same time, compared with the original model, the number of parameters is reduced by 50.4%, and the amount of computation is reduced by 35.44%. In summary, the model satisfies the requirements of the UAV platform for accuracy and lightweight in small target detection tasks.

    参考文献
    相似文献
    引证文献
引用本文

陈文轩,杨风暴,李菠,王肖霞,吉琳娜.改进YOLOv5s的轻量化无人机航拍小目标识别算法[J].电子测量技术,2026,49(4):116-125

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-04-16
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告