The safety inspection system for the underside of vehicles demands both high detection speed and accuracy, necessitating a model with small parameters and high network efficiency. To address these issues and improve the real-time performance of under-vehicle safety inspection, the YOLOv8n object-detection algorithm was enhanced, resulting in the PSP-YOLO algorithm. Firstly, the improved PSP-YOLO network introduces a P2 small-object detection head, which is particularly effective in detecting tiny objects, such as small dangerous objects under vehicles, thereby significantly enhancing the ability to detect such small targets. Secondly, the Space-to-Depth Conv (SPD-Conv) module was introduced into the backbone network, which improves the detection performance of low-quality samples and small objects while ensuring high precision and accurate localization of small targets in images. Finally, the Partial Convolution (PConv) lightweight module was also added to the feature fusion network, effectively reducing the model’s parameters and computational load, conserving computing resources, and improving detection speed and accuracy. Experimental results demonstrate that the modified model increased mAP@0.5 by 3.9% and mAP@0.5-0.95 by 3.6% on the dataset used, reduced the number of parameters to 2.53 M, and achieved a detection speed of 104.6 f/s. Therefore, the improved YOLOv8n algorithm significantly enhances the detection performance of dangerous objects, meeting the high-precision and real-time requirements for vehicle underside safety inspections.
Read full abstract