Aiming at the problems of large memory space occupied by remote sensing image detection models in complex scenes, high computational complexity, and detection speed that is difficult to meet the real-time target detection requirements, a model compression algorithm based on dynamic detection head and channel pruning is proposed. Based on YOLOv5s, the head of the original YOLOv5 is replaced by the DyHead detection head, which improves the detection performance of the model by combining and unifying the three attention mechanisms of scale, space and task. Secondly, the Focaler-IoU loss function is introduced to further improve the predictive regression ability of the model. Finally, the channel pruning algorithm is used to eliminate the redundant parameters in the network to further compress the parameters and computation of the model, and the accuracy of the pruned model is improved by model fine-tuning. The experimental results show that there is a large improvement on the DOTAV1.0 dataset, and the validity is further verified on the RSOD dataset.

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

Lightweight Remote Sensing Target Detection Algorithm Based on Improved YOLOv5

  • Chaomin Wang,
  • Fang Dong,
  • Sheng Liang,
  • Wanxin Yu,
  • Kaixian Wu,
  • Long Yu,
  • Wang Wenfeng

摘要

Aiming at the problems of large memory space occupied by remote sensing image detection models in complex scenes, high computational complexity, and detection speed that is difficult to meet the real-time target detection requirements, a model compression algorithm based on dynamic detection head and channel pruning is proposed. Based on YOLOv5s, the head of the original YOLOv5 is replaced by the DyHead detection head, which improves the detection performance of the model by combining and unifying the three attention mechanisms of scale, space and task. Secondly, the Focaler-IoU loss function is introduced to further improve the predictive regression ability of the model. Finally, the channel pruning algorithm is used to eliminate the redundant parameters in the network to further compress the parameters and computation of the model, and the accuracy of the pruned model is improved by model fine-tuning. The experimental results show that there is a large improvement on the DOTAV1.0 dataset, and the validity is further verified on the RSOD dataset.