Acta Optica Sinica, Volume. 40, Issue 12, 1215001(2020)

Traffic Light Detection Based on Optimized YOLOv3 Algorithm

Yingchun Sun, Shuguo Pan*, Tao Zhao, Wang Gao, and Jiansheng Wei
Author Affiliations
  • School of Instrument Science and Engineering, Southeast University, Nanjing, Jiangsu 210096, China
  • show less

    To solve the problems of high missed-detection rate and low recall rate existed in the YOLOv3 algorithm for detecting traffic lights, a traffic light detection method based on the optimized YOLOv3 algorithm is proposed. First, the K-means algorithm is used to cluster the data. By combining the clustering results with the statistical results of traffic light labels, the number and the width-height ratios of the prior boxes are determined. Then, the network structure is simplified according to the size characteristics of traffic lights. The 8× downsampling information and the 16× downsampling information are fused with high-level semantic information, and the object feature detection layer is established on two scales. Meanwhile, to avoid the disappearance problem of traffic light features with the deepening of the network, two sets of convolution layers are reduced before two object-detection layers, and thus the feature extraction steps are simplified. Finally, in the loss function, Gaussian distribution characteristics are used to evaluate the accuracy of the boundary box to improve the precision of traffic light detection. The experimental results reveal that the detection speed of the optimized YOLOv3 algorithm can reach 30 frames/s and the average precision is 9 percent higher than that of the original network, which effectively completes the detection of traffic lights.

    Tools

    Get Citation

    Copy Citation Text

    Yingchun Sun, Shuguo Pan, Tao Zhao, Wang Gao, Jiansheng Wei. Traffic Light Detection Based on Optimized YOLOv3 Algorithm[J]. Acta Optica Sinica, 2020, 40(12): 1215001

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Machine Vision

    Received: Dec. 31, 2019

    Accepted: Mar. 16, 2020

    Published Online: Jun. 3, 2020

    The Author Email: Pan Shuguo (psg@seu.edu.cn)

    DOI:10.3788/AOS202040.1215001

    Topics