Aiming at the problem of small target detection of unmanned aerial vehicle (UAV) platform with limited computing resources, a lightweight small target detection algorithm for UAV platforms based on YOLOv5 is proposed. The algorithm retains the original FPN + PAN and YOLOHead of YOLOv5 as the Neck network and Head network, respectively. Firstly, the original CSPDarknet53 structure of Backbone is lightweight, the original Focus layer is replaced with the Conv_BN_SiLu layer, and the Resblock body structure is replaced with the Inverted Resblock structure. Secondly, DSConv is used to lighten the Neck and reduce the number of the model model’s parameters. Finally, SPD-Conv is used to convert features from space to depth, strengthen the low-resolution features, and improve the model’s small targets detected ability. Results of the experiments show that the proposed algorithm achieves a good balance between the number of model parameters and the accuracy.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Small Target Detection Algorithm Based on Lightweighted YOLOv5

  • Shaowei Cao,
  • Lingxia Mu,
  • Youmin Zhang,
  • Ban Wang,
  • Yulong Zhang,
  • Xianghong Xue

摘要

Aiming at the problem of small target detection of unmanned aerial vehicle (UAV) platform with limited computing resources, a lightweight small target detection algorithm for UAV platforms based on YOLOv5 is proposed. The algorithm retains the original FPN + PAN and YOLOHead of YOLOv5 as the Neck network and Head network, respectively. Firstly, the original CSPDarknet53 structure of Backbone is lightweight, the original Focus layer is replaced with the Conv_BN_SiLu layer, and the Resblock body structure is replaced with the Inverted Resblock structure. Secondly, DSConv is used to lighten the Neck and reduce the number of the model model’s parameters. Finally, SPD-Conv is used to convert features from space to depth, strengthen the low-resolution features, and improve the model’s small targets detected ability. Results of the experiments show that the proposed algorithm achieves a good balance between the number of model parameters and the accuracy.