A deep learning method for traffic light status recognition  

作　　者：Lan Yang Zeyu He Xiangmo Zhao Shan Fang Jiaqi Yuan Yixu He Shijie Li Songyan Liu 

机构地区：[1]School of Information Engineering,Chang’an University,Xi’an 710064,China

出　　处：《Journal of Intelligent and Connected Vehicles》2023年第3期173-182,共10页智能网联汽车（英文）

基　　金：supported by the National Key R&D Program of China(2021YFB2501200);the Key Program of the National Natural Science Foundation of China(52131204);the Shaanxi Province Key Research and Development Program(2022GY-300).

摘　　要：Real-time and accurate traffic light status recognition can provide reliable data support for autonomous vehicle decision-making and control systems.To address potential problems such as the minor component of traffic lights in the perceptual domain of visual sensors and the complexity of recognition scenarios,we propose an end-to-end traffic light status recognition method,ResNeSt50-CBAM-DINO(RC-DINO).First,we performed data cleaning on the Tsinghua-Tencent traffic lights(TTTL)and fused it with the Shanghai Jiao Tong University’s traffic light dataset(S2TLD)to form a Chinese urban traffic light dataset(CUTLD).Second,we combined residual network with split-attention module-50(ResNeSt50)and the convolutional block attention module(CBAM)to extract more significant traffic light features.Finally,the proposed RC-DINO and mainstream recognition algorithms were trained and analyzed using CUTLD.The experimental results show that,compared to the original DINO,RC-DINO improved the average precision(AP),AP at intersection over union(IOU)=0.5(AP50),AP for small objects(APs),average recall(AR),and balanced F score(F1-Score)by 3.1%,1.6%,3.4%,0.9%,and 0.9%,respectively,and had a certain capability to recognize the partially covered traffic light status.The above results indicate that the proposed RC-DINO improved recognition performance and robustness,making it more suitable for traffic light status recognition tasks.

关 键 词：traffic light status recognition autonomous vehicle detection transformer with improved denoising anchor boxes(DINO) Chinese urban traffic light dataset(CUTLD) 

分 类 号：U491.5[交通运输工程—交通运输规划与管理]