Research on night-time vehicle target detection based on improved KSC-YOLO V5
Jun Guo,MengZhen Shao,XinYu Chen,Yang Yang,EnHui Sun
DOI: https://doi.org/10.1007/s11760-024-03576-5
IF: 1.583
2024-12-07
Signal Image and Video Processing
Abstract:In challenging night-time settings, drivers often struggle to accurately discern nearby vehicles, which can lead to traffic incidents due to complex lighting conditions and the issues of dim visibility, blurring, and occlusion of vehicles. To address this issue, this study introduces an advanced method that utilizes an enhanced YOLOv5 algorithm designed to improve the detection of vehicles during night driving and ensure detection accuracy in low-light environments. The method integrates SKAttention into the network's Backbone layer to address overlapping and redundant information within the model. This integration is achieved by generating multiple pathways with varying convolutional kernel sizes, which correspond to the different sensory field sizes of neurons, thus enhancing information processing efficiency. Additionally, the original up-sampling operator of the model is replaced with the Content-Aware Feature Reorganization (CARAFE), which leverages underlying content information to predict the reorganization kernel and reassemble predefined neighborhood features, preserving the quality of feature maps post-zooming. The incorporation of SIOU also significantly enhances the model's ability to pinpoint defect locations with greater accuracy, leading to improvements in both training speed and inference accuracy. Experimental results demonstrate that the optimized model, dubbed KSC-YOLOv5, surpasses the baseline YOLOv5 model, showing a mean average precision (mAP) increase of 2 percentage points to 91.2%. Furthermore, when compared with other iterations of the YOLO algorithm, KSC-YOLOv5 exhibits superior performance in terms of average detection accuracy. This refined model effectively navigates the challenges posed by inadequate lighting and vehicle blurring in nocturnal environments, achieving precise vehicle detection.
engineering, electrical & electronic,imaging science & photographic technology