Detection and Recognition for Autonomous Driving Based on Improved YOLO Algorithm
Jie Zhao,Chao Han,Yuan Ge
DOI: https://doi.org/10.23919/CCC63176.2024.10661633
2024-07-28
Abstract:Facing the most challenging topic of Autonomous Driving technology today, vision-based object detection is an essential component. In view of the characteristics of complex scenes, such as small targets, occlusion, and long distances, an improved YOLOv5-based vehicle view object detection model LampRepSE-YOLO, is proposed in this paper. Firstly, We designed the SE-PAN network to enhance information fusion pathways, incorporating SE attention mechanisms, and re-engineered the PANet structure; the multidimensional modeling of image feature channel relationships was realized; and the extraction effect of complex image features was improved. Secondly, because vehicles typically have limited onboard computing resources, a small-sized detection model is required. Through the application of channel-based LAMP, we achieved an impressive 48% reduction in parameters within the model, and the RepVGG algorithm is used in the Backbone part for structural reparametrization to increase the model complexity and pull up the model capacity at the same time, it doesn’t increase the GFLOPs. Third, in order to drive safely and improve detection accuracy, this paper innovatively uses multi-scale images for training. Finally, the KITTI dataset was selected as the training, validation, and testing set to conduct the index quantification experiment. The experimental results show that the mAP of LampRepSE-YOLO reaches 87.1%. which is higher than the YOLOv5 model 2.1%.
Computer Science,Engineering