+高级检索
基于多尺度融合和轻量化网络的无人机目标检测算法
DOI:
作者:
作者单位:

作者简介:

通讯作者:

基金项目:


Drone Target Detection Algorithm Based on Multi-scale Fusion and Lightweight Network
Author:
Affiliation:

Fund Project:

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

    针对在游乐场、公园等公共安全区域因无人机的尺度变化实时检测困难和计算资源有限的问题,提出一种网络动态实时检测无人机方法YOLO-Ads,以增加网络对无人机尺度变化的鲁棒性.首先自主构建了无人机数据集;其次将轻量化网络作为主干建立一个新的MDDRDNet网络,减小模型计算的复杂度,并且引入协调注意力机制模块,加强网络对空间和通道的关注度;然后采用均值聚类算法,重新生成先验框,在先验框的选择上结合多探测头和多数据集的寻优办法,使重新生成的先验框与无人机更加匹配;然后基于特征融合和残差思想建立一个新的探测头以适应更小尺度无人机的检测;最后,在检测模块中引入类激活映射模块生成热力图,以观察网络对无人机尺度变化的敏感程度,同时与当前主流网络SSD、CenterNet、YOLOv5、YOLOx等和不同主干网络ResNet、EfficientNet、VGGNet等进行对比实验.实验结果表明,新提出的算法在尺度变化的无人机检测上平均精度达到96.62%,相较YOLOv4算法提高了1.88%;检测速度为每秒47帧,相较YOLOv4算法提高了19帧;模型所占内存大约为10.844 M,约为原内存的六分之一,体现了该方法的有效性和鲁棒性.

    Abstract:

    Aiming at the difficulty of real-time detection and limited computing resources due to the scale change of drones in public safety areas such as playgrounds and parks, a network dynamic real-time detection method for drones, YOLO-Ads, is proposed to increase the robustness of network ability to detect drone change. Firstly, the drone data set was built independently. Secondly, a new MDDRDNet network was established with the lightweight network as the backbone to reduce the complexity of model calculation, and the coordinated attention mechanism module was introduced to strengthen the network’s attention to space and channels. Then, the mean clustering algorithm is used to regenerate the prior frame, and the optimization method combining multiple probes and multiple data sets is used in the selection of the prior frame, so that the regenerated prior frame matches the drone better. The idea of feature fusion and residual error establishes a new detector head to adapt to the detection of smaller-scale drones. Finally, a class activation mapping module is introduced into the detection module to generate a heat map, so as to observe the sensitivity of the network to changes in the scale of drones. At the same time, comparative experiments are conducted with the current mainstream networks SSD, CenterNet, YOLOv5, YOLOx, etc., and different backbone networks ResNet, EfficientNet, VGGNet, etc. The experimental results show that the newly proposed algorithm has an average accuracy of 96.62% in the detection of scale-changing drones. Compared with the YOLOv4 algorithm, it is increased by 1.88% . The detection speed is 47 frames per second, which is 19 frames higher than that of the YOLOv4 algorithm. The memory occupied by the model is about 10.844 M, which is about one-sixth of the original memory. It reflects the timeliness and robustness of the method.

    参考文献
    相似文献
    引证文献
文章指标
  • PDF下载次数:
  • HTML阅读次数:
  • 摘要点击次数:
  • 引用次数:
引用本文

薛珊 ?,卢涛 ,吕琼莹 ,曹国华 .基于多尺度融合和轻量化网络的无人机目标检测算法[J].湖南大学学报:自然科学版,2023,(8):82~93

复制
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-08-29
  • 出版日期:
作者稿件一经被我刊录用,如无特别声明,即视作同意授予我刊论文整体的全部复制传播的权利,包括但不限于复制权、发行权、信息网络传播权、广播权、表演权、翻译权、汇编权、改编权等著作使用权转让给我刊,我刊有权根据工作需要,允许合作的数据库、新媒体平台及其他数字平台进行数字传播和国际传播等。特此声明。
关闭