Traffic sign recognition based on deep convolutional neural network

Abstract

Traffic sign recognition (TSR) is an important component of automated driving systems. It is a rather challenging task to design a high-performance classifier for the TSR system. In this paper, we propose a new method for TSR system based on deep convolutional neural network. In order to enhance the expression of the network, a novel structure (dubbed block-layer below) which combines network-in-network and residual connection is designed. Our network has 10 layers with parameters (block-layer seen as a single layer): the first seven are alternate convolutional layers and block-layers, and the remaining three are fully-connected layers. We train our TSR network on the German traffic sign recognition benchmark (GTSRB) dataset. To reduce overfitting, we perform data augmentation on the training images and employ a regularization method named “dropout”. The activation function we employ in our network adopts scaled exponential linear units (SELUs), which can induce self-normalizing properties. To speed up the training, we use an efficient GPU to accelerate the convolutional operation. On the test dataset of GTSRB, we achieve the accuracy rate of 99.67%, exceeding the state-of-the-art results.

Keywords

A

References

[1] Nguwi Y.Y., Kouzani A.Z.. Neural Computing and Applications. 2008, 17: 265

[2] Mogelmose A., Trivedi M.M., Moeslund T.B.. IEEE Transactions on Intelligent Transportation Systems. 2012, 13: 1484

[3] Lu X., Wang Y., Zhou X., Zhang Z., Ling Z.. IEEE Transactions on Intelligent Transportation Systems. 2017, 18: 960

[4] Lim K.H., Seng K.P., Ang L.M.. Intra Color-shape Classification for Traffic Sign Recognition. 2010642

[5] Madani A., Yusof R.. Neural Computing and Applications. 20171

[6] Lau M.M., Lim K.H., Gopalai A.A.. Malaysia Traffic Sign Recognition with Convolutional Neural Network. 20151006

[7] Glorot X., Bordes A., Bengio Y.. Deep Sparse Rectifier Neural Networks. 2011315

[8] Maas A.L., Hannun A.Y., Ng A.Y.. Rectifier Nonlinearities Improve Neural Network Acoustic Models. 2013, 28: 6

[9] Xu B., Wang N., Chen T., Li M.. Empirical Evaluation of Rectified Activations in Convolutional Network. 2015

[10] Klambauer G., Unterthiner T., Mayr A., Hochreiter S.. Self-Normalizing Neural Networks. 2017

[11] LeCun Y., Bottou L., Bengio Y., Haffner P.. Proceedings of the IEEE. 1998, 86: 2278

[12] Krizhevsky A., Sutskever I., Hinton G.E.. Advances in Neural Information Processing Systems. 2012, 25: 1097

[13] Simonyan K., Zisserman A.. Very Deep Convolutional Networks for Large-scale Image Recognition. 2015

[14] Lin M., Chen Q., Yan S.. Network in Network. 2014

[15] Szegedy C., Liu W., Jia Y., Sermanet P., Reed S., Anguelov D., Erhan D., Vanhoucke V., Rabinovich A.. Going Deeper with Convolutions. 20151

[16] He K., Zhang X., Ren S., Sun J.. Deep Residual Learning for Image Recognition. 2016770

[17] Szegedy C., Ioffe S., Vanhoucke V., Alemi A.A.. Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning. 20174278

[18] Stallkamp J., Schlipsing M., Salmen J., Igel C.. Neural Networks. 2012, 32: 323

[19] Abadi M., Agarwal A., Barham P., Brevdo E.. Tensorflow: Large-scale Machine Learning on Heterogeneous Distributed Systems. 2016

[20] Kingma D., Ba J.. Adam: A Method for Stochastic Optimization. 2014

[21] Hinton G.E., Srivastava N., Krizhevsky A., Sutskever I., Salakhutdinov R.R.. Improving Neural Networks by Preventing Co-adaptation of Feature Detectors. 2012

[22] Glorot X., Bengio Y.. Understanding the Difficulty of Training Deep Feedforward Neural Networks. 2010249