Improved traffic signal light recognition algorithm based on YOLO v3

Feng Yu; Mingen Zhong; Shifu Tang; Zhonggang Zheng

doi:10.1117/12.2634502

12 May 2022 Improved traffic signal light recognition algorithm based on YOLO v3

Feng Yu, Mingen Zhong, Shifu Tang, Zhonggang Zheng

Proceedings Volume 12173, International Conference on Optics and Machine Vision (ICOMV 2022); 1217311 (2022) https://doi.org/10.1117/12.2634502
Event: International Conference on Optics and Machine Vision (ICOMV 2022), 2022, Guangzhou, China

Abstract

Aiming at the problems of misdetection and missed detection in traffic signal detection tasks in current traffic scenes, this paper proposes an improved YOLO v3 traffic signal recognition algorithm YOLO v3-SimAM. First, introduce the new attention mechanism SimAM to enhance the features; secondly, add the SSH network structure after the feature layer extracted by the FPN feature pyramid to expand the deep network receptive field to further strengthen the feature extraction; finally, the last convolution of YOLO Head is replaced with dynamic convolution to improve the accuracy of target frame detection. At the same time, in the loss function design, the original intersection ratio (IoU) loss function is replaced with the DIoU loss function, and finally the YOLO v3-SimAM algorithm is formed. The algorithm is applied to traffic signal detection tasks. On the public TTTL (Tsinghua-Tencent Traffic Light) data set, the average accuracy of the algorithm (mAP) index reached 67.67%, which is an increase of 1.2% compared with the YOLO v3 algorithm. The results show that compared with the original YOLO v3 algorithm, the YOLO v3-SimAM algorithm proposed in this paper can detect traffic lights more accurately.

Citation Download Citation

Feng Yu, Mingen Zhong, Shifu Tang, and Zhonggang Zheng "Improved traffic signal light recognition algorithm based on YOLO v3", Proc. SPIE 12173, International Conference on Optics and Machine Vision (ICOMV 2022), 1217311 (12 May 2022); https://doi.org/10.1117/12.2634502

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
8 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Target detection

Detection and tracking algorithms

Signal detection

Convolution

Neurons

Feature extraction

Data modeling

Show All Keywords

Keywords/Phrases

Search In:

Publication Years