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

1.河北省交通安全与控制重点实验室 石家庄 050043; 2.石家庄铁道大学交通运输学院 石家庄 050043; 3.沧州渤海新区黄骅市交通运输局 黄骅 061100

作者简介:

通讯作者:

中图分类号:

TP391.41;V19

基金项目:

中央引导地方科技发展资金(226Z6101G)、河北省及石家庄市引进国外智力项目(20212023)、石家庄市科技计划项目(221130134A)资助


UAV aerial object detection algorithm based on improved YOLOv4
Author:
Affiliation:

1.Key Laboratory of Traffic Safety and Control of Hebei Province,Shijiazhuang 050043,China; 2.School of Traffic and Transportation, Shijiazhuang Tiedao University,Shijiazhuang 050043,China; 3.Cangzhou Bohai New Area Huanghua Transportation Bureau,Huanghua 061100,China

Fund Project:

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

    针对无人机航拍目标检测对检测速度的高要求以及航拍图像小目标较多时易出现漏检、误检的问题,提出一种基于改进YOLOv4的无人机航拍目标检测算法。首先,引入轻量级网络MobileNetv3替换YOLOv4的主干特征提取网络,并采用深度可分离卷积替换网络其余部分的3×3标准卷积,降低了模型复杂度,提升了检测速度;其次,增加了针对小目标的104×104的浅层检测层,将原特征提取网络的3种检测尺度扩展为4种,对应加深特征融合网络层数,提升了算法对小目标的检测精度;最后,采用K-means++聚类算法重新设计了初始锚框,加快了网络的收敛速度。在无人机航拍数据集上进行了对比实验,结果表明所提算法与原算法相比,在保证平均检测精度的同时,提升了小目标检测精度,且模型参数量减少了60%,检测速度提升了15.2%,在实时性和准确性方面均有较好性能。

    Abstract:

    A UAV aerial object detection algorithm based on modified YOLOv4 is proposed in order to address the high need of detection speed for UAV aerial object detection as well as the issue of missed detection and false detection when there are many small targets in aerial images. Firstly, the lightweight network MobileNetv3 is introduced to replace the main feature extraction network of YOLOv4, and the depth separable convolution is used to replace the 3×3 standard convolution of the network, which reduces the complexity of the model and improves the detection speed. Secondly, the 104×104 shallow detection layer for small targets is added, the three detection scales of the original feature extraction network are increased to four, and the number of feature fusion network layers is increased. These changes increase the algorithm′s accuracy in detecting small targets. Finally, the K-means++clustering technique is used to redesign the initial anchor frame, accelerating the network′s rate of convergence. The UAV aerial data set is used in a comparison experiment. The findings demonstrate that as compared to the original approach, the suggested technique not only ensures average detection accuracy but significantly enhances the detection accuracy of small targets. The detection time is 15.2% faster while the model parameters are lowered by 60%. It performs accurately and with good real-time performance.

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

赵耘彻,张文胜,刘世伟.基于改进YOLOv4的无人机航拍目标检测算法[J].电子测量技术,2023,46(8):169-175

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-02-07
  • 出版日期: