Multimedia Tools and Applications
https://doi.org/10.1007/s11042-020-08722-y
Real-time traffic sign detection and classification
towards real traffic scene
Yiqiang Wu
1,2
· Zhiyong Li
1,2
· Ying Chen
1,2
· Ke Nai
1,2
· Jin Yuan
1,2
Received: 7 May 2019 / Revised: 4 December 2019 / Accepted: 31 January 2020 /
© Springer Science+Business Media, LLC, part of Springer Nature 2020
Abstract
In this paper we propose a real-time traffic sign recognition algorithm which is robust to
the small-sized objects and can identify all traffic sign categories. Specifically, we present a
two-level detection framework which consists of the region proposal module(RPM) which
is responsible for locating the objects and the classification module(CM) which aims to clas-
sify the located objects. In addition, to solve the problem of insufficient samples, we present
an effective data augmentation method based on traffic sign logo to generate enough train-
ing data. The experiments are conducted in TT100k, and the results show the superiority of
our method.
Keywords Traffic sign recognition · Small object detection · Data augmentation ·
Image synthesis
1 Introduction
Traffic sign recognition is an important sub-task in the advanced driver assistant systems
and autonomous driving systems. In general, there are two stages in a traffic sign recog-
nition system: finding the locations of the traffic signs in real traffic scenes (traffic sign
detection) and classifying the detected traffic signs into their specific sub-classes (traffic
sign classification). Traffic sign detection faces many difficulties in real traffic scenes due
to illumination changes, partial occlusion, cluttered background, and small size, as shown
in Fig. 1. And in the second stage, there are also many problems, such as missing samples,
unbalanced sample categories, the rare sample numbers etc. TT100k [37] is a Chinese traf-
fic sign dataset and contains over 150 categories, which is the most widely covered traffic
sign dataset in the current. However, many rather rare traffic signs are still not included.
And not only that, the existing categories in the dataset are extremely imbalanced, let alone
Zhiyong Li
zhiyong.li@hnu.edu.cn
1
College of Computer Science and Electronic Engineering Hunan University, Changsha, China
2
Key Laboratory for Embedded and Network Computing of Hunan Province, Changsha, China