Acta Optica Sinica, Volume. 40, Issue 12, 1215001(2020)
Traffic Light Detection Based on Optimized YOLOv3 Algorithm
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.
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
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)