Research on Improved YOLOv5 Vehicle Target Detection Algorithm in Aerial Images

Autor: Xue Yang, Jihong Xiu, Xiaojia Liu
Jazyk: angličtina
Rok vydání: 2024
Předmět:
Zdroj: Drones, Vol 8, Iss 5, p 202 (2024)
Druh dokumentu: article
ISSN: 2504-446X
DOI: 10.3390/drones8050202
Popis: Aerial photoelectric imaging payloads have become an important means of reconnaissance and surveillance in recent years. However, aerial images are easily affected by external conditions and have unclear edges, which greatly reduces the accuracy of imaging target recognition. This paper proposes the M-YOLOv5 model, which uses a shallow feature layer. The RFBs module is introduced to improve the receptive field and detection effect of small targets. In the neck network part, the BiFPN structure is used to reuse the underlying features to integrate more features, and a CBAM attention mechanism is added to improve detection accuracy. The experimental results show that the detection effect of this method on the DroneVehicle dataset is better than that of the original network, with the precision rate increased by 2.8%, the recall rate increased by 16%, and the average precision increased by 2.3%. Considering the real-time problem of target detection, based on the improved model, the Clight-YOLOv5 model is proposed, by lightweighting the network structure and using the depth-separable convolution optimization module. After lightweighting, the number of model parameters is decreased by 71.3%, which provides a new idea for lightweight target detection and proves the model’s effectiveness in aviation scenarios.
Databáze: Directory of Open Access Journals