Abstract
Self-driving vehicles are being tested to make them more road-ready and safer for a real traffic environment. Automobile giants: like Tesla, Waymo, Toyota, etc., are working exhaustively to cater to the needs of futuristic smart vehicles using deep learning methodologies. For a self-driving car to avoid collision situation, it must be able to accurately detect and classify traffic lights. After performing exhaustive experiments, we chose to compare the feature extraction capabilities of various pretrained CNN-based transfer learning models like VGG16, ResNet50, AlexNet, DenseNet121, InceptionV3, and Xception on freely available Lara & Lisa traffic light datasets. We segregated the Lisa traffic light dataset into day and night subsets and then manually separated the images into various traffic light classes like dayRed, dayYellow, nightYellow, nightGreen, road and traffic lights, LeftGreenArrow, and RightGreenArrow etc. We used the random forest classifier to identify the color of the detected traffic light. DenseNet121 achieved a top accuracy of 100% on the Lara traffic light dataset and the maximum accuracy of 99.88% on the Lisa Day-light dataset and 98.89% accuracy on the Lisa Night-light dataset.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Breiman, L.: Random forests. Mach. Learning 45(1), 5–32 (2001)
Schuldt, C., Laptev, I., Caputo, B.: Recognizing human actions: a local SVM approach. In Proceedings of the 17th International Conference on Pattern Recognition, 2004. ICPR 2004, vol. 3, pp. 32–36, IEEE, (Aug 2004)
Chen, T., Guestrin, C.: Xgboost: a scalable tree boosting system. In Proceedings of the 22nd ACM Sigkdd International Conference on Knowledge Discovery and Data Mining, pp. 785–794, (Aug 2016)
Simonyan, K., Zisserman, A.: Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556 (2014)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778, (2016)
Huang, G., Liu, Z., Van Der Maaten, L., Weinberger, K.Q.: Densely connected convolutional networks. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 4700–4708 (2017)
Jensen, M.B., Nasrollahi, K., Moeslund, T.B.: Evaluating state-of-the-art object detector on challenging traffic light data. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition Workshops, pp. 9–15 (2017)
Zhang, Y., Xue, J., Zhang, G., Zhang, Y., Zheng, N.: A multi-feature fusion based traffic light recognition algorithm for intelligent vehicles. In Proceedings of the 33rd Chinese Control Conference, pp. 4924–4929, IEEE (Jul 2014)
Possatti, L.C., Guidolini, R., Cardoso, V.B., Berriel, R.F., Paixão, T.M., Badue, C., Oliveira-Santos, T.: Traffic light recognition using deep learning and prior maps for autonomous cars. In 2019 International Joint Conference on Neural Networks (IJCNN), pp. 1–8, IEEE (Jul 2019)
Nienhüser, D., Drescher, M., Zöllner, J.M.: Visual state estimation of traffic lights using hidden Markov models. In 13th International IEEE Conference on Intelligent Transportation Systems, pp. 1705–1710, IEEE, (Sep 2010)
Lu, Y., Lu, J., Zhang, S., Hall, P.: Traffic signal detection and classification in street views using an attention model. Comput. Visual Media 4(3), 253–266 (2018)
Wang, J.G., Zhou, L.B.: Traffic light recognition with high dynamic range imaging and deep learning. IEEE Trans. Intell. Transp. Syst. 20(4), 1341–1352 (2018)
Jensen, M.B., Philipsen, M.P., Bahnsen, C., Møgelmose, A., Moeslund, T. B., Trivedi, M.M.: Traffic light detection at night: comparison of a learning-based detector and three model-based detectors. In International Symposium on Visual Computing, pp. 774–783, Springer, Cham (Dec 2015)
Tran, T.H.P., Jeon, J.W.: Accurate real-time traffic light detection using YOLOv4. In 2020 IEEE International Conference on Consumer Electronics-Asia (ICCE-Asia), pp. 1–4, IEEE (Nov 2020)
Hein, D.: Traffic Light Detection with Convolutional Neural Networks and 2D Camera Data, Doctoral dissertation, fu-berlin, (2020)
Chollet, F.: Xception: deep learning with depthwise separable convolutions. Proc 30th IEEE Conf. Comput. Vis. Pattern Recognition, CVPR 2017 2017-Janua 1800–7 (2017)
Widyantoro, D.H.: Circle and arrow traffic light recognition. In 2017 IEEE International Conference on Cybernetics and Computational Intelligence (CyberneticsCom), pp. 34–38, IEEE (Nov 2017)
Greenhalgh, J., Mirmehdi, M.: Traffic sign recognition using MSER and random forests. In 2012 Proceedings of the 20th European Signal Processing Conference (EUSIPCO), pp. 1935–1939, IEEE (Aug 2012)
Stallkamp, J., Schlipsing, M., Salmen, J., Igel, C.: Man vs. computer: benchmarking machine learning algorithms for traffic sign recognition. Neural Netw. 32, 323–332 (2012)
Lin, C., Li, L., Luo, W., Wang, K.C., Guo, J.: Transfer learning based traffic sign recognition using inception-v3 model. Period. Polytech. Transp. Eng. 47(3), 242–250 (2019)
Author information
Authors and Affiliations
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Gautam, S., Kumar, A. (2022). Automatic Traffic Light Detection for Self-Driving Cars Using Transfer Learning. In: Nagar, A.K., Jat, D.S., MarÃn-Raventós, G., Mishra, D.K. (eds) Intelligent Sustainable Systems. Lecture Notes in Networks and Systems, vol 333. Springer, Singapore. https://doi.org/10.1007/978-981-16-6309-3_56
Download citation
DOI: https://doi.org/10.1007/978-981-16-6309-3_56
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-16-6308-6
Online ISBN: 978-981-16-6309-3
eBook Packages: Intelligent Technologies and RoboticsIntelligent Technologies and Robotics (R0)