Lightweight Remote Sensing Target Detection Algorithm Based on Improved YOLOv5
摘要
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.