弱光照条件下交通标志检测与识别

Traffic signs detection and recognition under low-illumination conditions

  • 摘要: 针对弱光照条件下交通标志易发生漏检和定位不准的问题,本文提出了增强YOLOv3(You only look once)检测算法,一种实时自适应图像增强与优化YOLOv3网络结合的交通标志检测与识别方法。首先构建了大型复杂光照中国交通标志数据集;然后针对复杂的弱光照图像提出自适应增强算法,通过调整图像亮度和对比度强化交通标志与背景之间的差异;最后采用YOLOv3网络框架检测交通标志。为了降低先验锚点框设置精度以及图像中背景与前景比例严重失衡对检测精度造成的影响,优化了先验锚点框聚类算法和网络的损失函数。对比实验结果表明,在实时性大致相当的情况下,本文提出的增强YOLOv3检测算法较标准YOLOv3算法对交通标志有更高的回归精度和置信度,召回率和准确率分别提高0.96%和0.48%。

     

    Abstract: Traffic sign detection and recognition, which are important to ensure traffic safety, have been a research hotspot. In recent years, with the rapid development of automated driving technology, significant progress has been made in developing more accurate and efficient deep learning algorithms for traffic sign detection and recognition. However, these studies mainly focus on foreign traffic signs and do not consider the low-illumination conditions in practical application, which is a common scene. Therefore, many challenges still exist in the application of traffic sign detection and recognition in traffic scenes. To solve the problems of easy omission and inaccurate positioning for traffic sign detection and recognition under complex illumination conditions, the enhanced YOLOv3 (You only look once) detection algorithm, a traffic sign detection and recognition method combining real-time adaptive image enhancement and the YOLOv3 frame was proposed. First, a large and complex illumination traffic sign dataset for Chinese traffic was constructed; it included globally low illumination, locally low illumination, and sufficient illumination images. Then an adaptive enhancement algorithm was proposed for low-illumination images, which can enhance the difference between traffic signs and background by adjusting the brightness and contrast of the images. Finally, high-quality and discrimination images as input were transmitted to the YOLOv3 network framework, and traffic sign detection and recognition were performed. To reduce the influences of the prior anchor box setting accuracy and the imbalance between the background and foreground on the detection accuracy, the clustering algorithm for the prior anchor box and loss function for the network were optimized. The results of the comparison experiment with the LISA dataset and complex illumination traffic sign dataset for Chinese traffic show that the proposed enhanced YOLOv3 detection algorithm has higher regression accuracy and category confidence than the published YOLOv3 algorithm for traffic signs; the recall and precision are higher by 0.96% and 0.48%, respectively, which indicates the application potential of the proposed algorithm in actual traffic scenarios.

     

/

返回文章
返回