Traffic Light Detection by Integrating Feature Fusion and Attention Mechanism

Chi Hung Chuang, Chun Chieh Lee, Jung Hua Lo, Kuo Chin Fan

研究成果: 雜誌貢獻期刊論文同行評審


Path planning is a key problem in the design of autonomous driving systems, and accurate traffic light detection is very important for robust routing. In this paper, we devise an object detection model, which mainly focuses on traffic light classification at a distance. In the past, most techniques employed in this field were dominated by high-intensity convolutional neural networks (CNN), and many advances have been achieved. However, the size of traffic lights may be small, and how to detect them accurately still deserves further study. In the object detection domain, the scheme of feature fusion and transformer-based methods have obtained good performance, showing their excellent feature extraction capability. Given this, we propose an object detection model combining both the pyramidal feature fusion and self-attention mechanism. Specifically, we use the backbone of the mainstream one-stage object detection model consisting of a parallel residual bi-fusion (PRB) feature pyramid network and attention modules, coupling with architectural tuning and optimizer selection. Our network architecture and module design aim to effectively derive useful features aimed at detecting small objects. Experimental results reveal that the proposed method exhibits noticeable improvement in many performance indicators: precision, recall, F1 score, and mAP, compared to the vanilla models. In consequence, the proposed method obtains good results in traffic light detection.

期刊Electronics (Switzerland)
出版狀態已出版 - 9月 2023


深入研究「Traffic Light Detection by Integrating Feature Fusion and Attention Mechanism」主題。共同形成了獨特的指紋。