Structure reparameterized YOLOV5s for improved traffic light detection

Peilin Liu; Zhaoyang Xie; Taijun Li

doi:10.1117/12.2681673

1 June 2023 Structure reparameterized YOLOV5s for improved traffic light detection

Peilin Liu, Zhaoyang Xie, Taijun Li

Proceedings Volume 12718, International Conference on Cyber Security, Artificial Intelligence, and Digital Economy (CSAIDE 2023); 1271824 (2023) https://doi.org/10.1117/12.2681673
Event: International Conference on Cyber Security, Artificial Intelligence, and Digital Economy (CSAIDE 2023), 2023, Nanjing, China

Abstract

The detection and recognition of traffic signals, which are important elements in the driving environment, are part of the intelligent driving domain, and the results of the detection and recognition affect the driving safety of intelligent vehicles. However, the classic traffic light signal detection algorithm model is too large, which has certain performance requirements on the devices deployed at mobile terminals. In this paper, we propose a structurally reparameterized network structure that can achieve the performance of classical mainstream algorithms and also significantly reduce the number of parameters and operations of the model. Firstly, the C3 structure of the yolov5s backbone network was replaced with MobileOne structure. Depthwise Convolution was used to reduce parameters and operation costs on the neck part of the model. The CIOU which calculated the location loss was replaced with the EIOU which focused on the length loss and width loss of the anchor boxes. Finally, SE attention module is introduced to further improve the characterization ability of the model. Experiments show that compared with the original Yolov5s algorithm, the parameter number and GFLOPs of the improved structure reparameterized Yolov5s decrease by 29.8%. the map0.5 performance index increased by 0.7% and reasoning speed increased by 19.2%, which effectively achieved a more real-time effect on traffic signal light detection and has certain application value. It can provide effective support for the driving safety of smart cars.

Citation Download Citation

Peilin Liu, Zhaoyang Xie, and Taijun Li "Structure reparameterized YOLOV5s for improved traffic light detection", Proc. SPIE 12718, International Conference on Cyber Security, Artificial Intelligence, and Digital Economy (CSAIDE 2023), 1271824 (1 June 2023); https://doi.org/10.1117/12.2681673

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
6 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Object detection

Convolution

Detection and tracking algorithms

Evolutionary algorithms

Autonomous driving

Computer vision technology

Performance modeling

Show All Keywords

Keywords/Phrases

Search In:

Publication Years