Small Traffic Sign Detector in Real-time Based on Improved YOLO-v4

  • Tingting Yang
  • , Chao Tong*
  • *Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

As one of the indispensable parts of the smart transportation system, traffic sign detection is still an open challenge for the reason that traffic sign detection, one kind of small object detection, requires high precision, high reliability, and real-time speed. In this paper, a real-time detector for small traffic sign improved from YOLO-v4 is proposed. First, a transpose convolution layer is added into the backbone of YOLO-v4, named CSPDark-53, to get larger feature maps with richer semantic information of small objects. Secondly, aiming at fusing multi-scale feature maps more effectively, we present two novel attention modules, the upsampling attention module and the downsampling attention module, to calculate the channel weights and spatial masks to help the network focus on useful features and suppress invalid features. The experiments show our detector outperforms the compared models in Chinese Traffic sign dataset TT100k.

Original languageEnglish
Title of host publication2021 IEEE 23rd International Conference on High Performance Computing and Communications, 7th International Conference on Data Science and Systems, 19th International Conference on Smart City and 7th International Conference on Dependability in Sensor, Cloud and Big Data Systems and Applications, HPCC-DSS-SmartCity-DependSys 2021
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages1318-1324
Number of pages7
ISBN (Electronic)9781665494571
DOIs
StatePublished - 2022
Event23rd IEEE International Conference on High Performance Computing and Communications, 7th IEEE International Conference on Data Science and Systems, 19th IEEE International Conference on Smart City and 7th IEEE International Conference on Dependability in Sensor, Cloud and Big Data Systems and Applications, HPCC-DSS-SmartCity-DependSys 2021 - Haikou, Hainan, China
Duration: 20 Dec 202122 Dec 2021

Publication series

Name2021 IEEE 23rd International Conference on High Performance Computing and Communications, 7th International Conference on Data Science and Systems, 19th International Conference on Smart City and 7th International Conference on Dependability in Sensor, Cloud and Big Data Systems and Applications, HPCC-DSS-SmartCity-DependSys 2021

Conference

Conference23rd IEEE International Conference on High Performance Computing and Communications, 7th IEEE International Conference on Data Science and Systems, 19th IEEE International Conference on Smart City and 7th IEEE International Conference on Dependability in Sensor, Cloud and Big Data Systems and Applications, HPCC-DSS-SmartCity-DependSys 2021
Country/TerritoryChina
CityHaikou, Hainan
Period20/12/2122/12/21

Keywords

  • YOLO-v4
  • attention module
  • small object detection
  • traffic sign detecion

Fingerprint

Dive into the research topics of 'Small Traffic Sign Detector in Real-time Based on Improved YOLO-v4'. Together they form a unique fingerprint.

Cite this