Wei Yongke, Zeng Zimu, He Tingquan, Yu Shanchuan, Du Yuchuan, Zhao Cong
Department of Geotechnical Engineering, College of Civil Engineering, Tongji University, Shanghai 200092, China.
State Key Laboratory for Disaster Reduction in Civil Engineering, Tongji University, Shanghai 200092, China.
Sensors (Basel). 2024 Sep 12;24(18):5912. doi: 10.3390/s24185912.
Vehicle detection is essential for road traffic surveillance and active safety management. Deep learning methods have recently shown robust feature extraction capabilities and achieved improved detection results. However, vehicle detection models often perform poorly under abnormal lighting conditions, especially in highway tunnels. We proposed an adaptive vehicle detection model that accounts for varying luminance intensities to address this issue. The model categorizes the image data into abnormal and normal luminance scenarios. We employ an improved CycleGAN with edge loss as the adaptive luminance adjustment module for abnormal luminance scenarios. This module adjusts the brightness of the images to a normal level through a generative network. Finally, YOLOv7 is utilized for vehicle detection. The experimental results demonstrate that our adaptive vehicle detection model effectively detects vehicles under abnormal luminance scenarios in highway tunnels. The improved CycleGAN can effectively mitigate edge generation distortion. Under abnormal luminance scenarios, our model achieved a 16.3% improvement in precision, a 1.7% improvement in recall, and a 9.8% improvement in mAP_0.5 compared to the original YOLOv7. Additionally, our adaptive luminance adjustment module is transferable and can enhance the detection accuracy of other vehicle detection models.
车辆检测对于道路交通监控和主动安全管理至关重要。深度学习方法最近展现出强大的特征提取能力,并取得了更好的检测结果。然而,车辆检测模型在异常光照条件下往往表现不佳,尤其是在公路隧道中。我们提出了一种自适应车辆检测模型,该模型考虑了不同的亮度强度来解决这一问题。该模型将图像数据分类为异常和正常亮度场景。对于异常亮度场景,我们采用带有边缘损失的改进型CycleGAN作为自适应亮度调整模块。该模块通过生成网络将图像亮度调整到正常水平。最后,利用YOLOv7进行车辆检测。实验结果表明,我们的自适应车辆检测模型能够在公路隧道的异常亮度场景下有效地检测车辆。改进后的CycleGAN能够有效减轻边缘生成失真。在异常亮度场景下,与原始的YOLOv7相比,我们的模型在精度上提高了16.3%,召回率提高了1.7%,mAP_0.5提高了9.8%。此外,我们的自适应亮度调整模块具有可迁移性,能够提高其他车辆检测模型的检测精度。