Liu Zhaohui, Yan Jun, Zhang Jinzhao
State Key Laboratory of Automotive Simulation and Control (ASCL), Changchun 130025, China.
College of Transportation, Shandong University of Science and Technology, Qingdao 266590, China.
Sensors (Basel). 2024 Jul 5;24(13):4370. doi: 10.3390/s24134370.
The efficient and accurate identification of traffic signs is crucial to the safety and reliability of active driving assistance and driverless vehicles. However, the accurate detection of traffic signs under extreme cases remains challenging. Aiming at the problems of missing detection and false detection in traffic sign recognition in fog traffic scenes, this paper proposes a recognition algorithm for traffic signs based on pix2pixHD+YOLOv5-T. Firstly, the defogging model is generated by training the pix2pixHD network to meet the advanced visual task. Secondly, in order to better match the defogging algorithm with the target detection algorithm, the algorithm YOLOv5-Transformer is proposed by introducing a transformer module into the backbone of YOLOv5. Finally, the defogging algorithm pix2pixHD is combined with the improved YOLOv5 detection algorithm to complete the recognition of traffic signs in foggy environments. Comparative experiments proved that the traffic sign recognition algorithm proposed in this paper can effectively reduce the impact of a foggy environment on traffic sign recognition. Compared with the YOLOv5-T and YOLOv5 algorithms in moderate fog environments, the overall improvement of this algorithm is achieved. The precision of traffic sign recognition of the algorithm in the fog traffic scene reached 78.5%, the recall rate was 72.2%, and mAP@0.5 was 82.8%.
交通标志的高效准确识别对于主动驾驶辅助和无人驾驶车辆的安全性和可靠性至关重要。然而,在极端情况下准确检测交通标志仍然具有挑战性。针对雾天交通场景中交通标志识别存在的漏检和误检问题,本文提出了一种基于pix2pixHD+YOLOv5-T的交通标志识别算法。首先,通过训练pix2pixHD网络生成去雾模型以满足高级视觉任务。其次,为了使去雾算法更好地与目标检测算法匹配,通过在YOLOv5的主干中引入Transformer模块提出了YOLOv5-Transformer算法。最后,将去雾算法pix2pixHD与改进后的YOLOv5检测算法相结合,完成雾天环境下交通标志的识别。对比实验证明,本文提出的交通标志识别算法能够有效降低雾天环境对交通标志识别的影响。与中等雾天环境下的YOLOv5-T和YOLOv5算法相比,该算法实现了整体提升。该算法在雾天交通场景中交通标志识别的精度达到78.5%,召回率为72.2%,mAP@0.5为82.8%。