Abstract
Traffic sign recognition is a popular task in the field of computer vision, but it faces challenges such as small object size, complex scenes, and real-time requirements. In this paper, a model framework is proposed to comprehensively improves the detection and recognition of traffic signs. To address small objects, FlexCut data augmentation is introduced, which generates non-repetitive sub-images through the strategy of maximizing sample region cropping. This approach enhances the detection ability of small objects. Additionally, PIoU loss function based on keypoints is also investigated, which accurately guides the position and shape of the bounding boxes by considering factors such as overlap area, distance, aspect ratio, and geometric size. Furthermore, the YOLOv5s network is enhanced by integrating the TransformerBlock module, SimAM attention mechanism, and Decoupled detection head to enhance the receptive field and feature extraction capability. In the experiments conducted on the TT100K dataset,, the proposed YOLOv5T achieves significant performance with an mAP@0.5 of 87.5% and an mAP@0.5:0.95 of 66.1%. These results validate the effectiveness of the proposed approach in addressing traffic sign recognition problems.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Ashish, V.: Attention is all you need. In: Advances Neural Information Processing system 30, I (2017)
Bochkovskiy, A., Wang, C.Y., Liao, H.Y.M.: Yolov4: optimal speed and accuracy of object detection. arXiv Preprint. arXiv:2004.10934 (2020)
Chen, C., et al.: RRNet: a hybrid detector for object detection in drone-captured images. In: Proceedings of the IEEE/CVF International Conference on Computer Vision Workshops (2019)
DeVries, T., Taylor, G.W.: Improved regularization of convolutional neural networks with cutout. arXiv preprint arXiv:1708.04552 (2017)
He, J., Erfani, S., Ma, X., Bailey, J., Chi, Y., Hua, X.S.: Alpha-IoU: a family of power intersection over union losses for bounding box regression. In: Advanced in Neural Information Processing Systems, vol. 34, pp. 20230–20242 (2021)
Leng, J., Ren, Y., Jiang, W., Sun, X., Wang, Y.: Realize your surroundings: exploiting context information for small object detection. Neurocomputing 433, 287–299 (2021)
Li, J., Liang, X., Wei, Y., Xu, T., Feng, J., Yan, S.: Perceptual generative adversarial networks for small object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 1222–1230 (2017)
Lim, J.S., Astrid, M., Yoon, H.J., Lee, S.I.: Small object detection using context and attention. In: 2021 International Conference on Artificial Intelligence in Information and Communication (ICAIIC), pp. 181–186. IEEE (2021)
Liu, Z., Gao, G., Sun, L., Fang, L.: IPG-Net: image pyramid guidance network for small object detection. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, pp. 1026–1027 (2020)
Rezatofighi, H., Tsoi, N., Gwak, J., Sadeghian, A., Reid, I., Savarese, S.: Generalized intersection over union: a metric and a loss for bounding box regression. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 658–666 (2019)
Riveiro, B., DÃaz-Vilariño, L., Conde-Carnero, B., Soilán, M., Arias, P.: Automatic segmentation and shape-based classification of retro-reflective traffic signs from mobile lidar data. IEEE J. Sel. Top. Appl. Earth Obs. Remote Sens. 9(1), 295–303 (2015)
Shi, Y., Li, X., Chen, M.: SC-YOLO: a object detection model for small traffic signs. IEEE Access 11, 11500–11510 (2023)
Singh, B., Davis, L.S.: An analysis of scale invariance in object detection snip. In: Proceedings of the IEEE Conference on Computer Vision and pattern Recognition, pp. 3578–3587 (2018)
Wang, L., Wang, L., Zhu, Y., Chu, A., Wang, G.: CDFF: a fast and highly accurate method for recognizing traffic signs. Neural Comput. Appl. 35(1), 643–662 (2023)
Yan, Z., Zheng, H., Li, Y., Chen, L.: Detection-oriented backbone trained from near scratch and local feature refinement for small object detection. Neural Process. Lett. 53, 1921–1943 (2021)
Yang, L., Zhang, R.Y., Li, L., Xie, X.: SimAM: a simple, parameter-free attention module for convolutional neural networks. In: International Conference on Machine Learning, pp. 11863–11874. PMLR (2021)
Yang, T., Long, X., Sangaiah, A.K., Zheng, Z., Tong, C.: Deep detection network for real-life traffic sign in vehicular networks. Comput. Netw. 136, 95–104 (2018)
Yang, Z., et al.: Prior knowledge guided small object detection on high-resolution images. In: 2019 IEEE International Conference on Image Processing (ICIP), pp. 86–90. IEEE (2019)
Yu, J., Jiang, Y., Wang, Z., Cao, Z., Huang, T.: UnitBox: an advanced object detection network. In: Proceedings of the 24th ACM International Conference on Multimedia, pp. 516–520 (2016)
Yuan, X., Kuerban, A., Chen, Y., Lin, W.: Faster light detection algorithm of traffic signs based on yolov5s-a2. IEEE Access, 19395–19404 (2022)
Yun, S., Han, D., Oh, S.J., Chun, S., Choe, J., Yoo, Y.: CutMix: regularization strategy to train strong classifiers with localizable features. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 6023–6032 (2019)
Zhang, H., Cisse, M., Dauphin, Y., Lopez-Paz, D.: Mixup: beyond empirical risk management. In: 6th International Conference Learning Representations (ICLR), pp. 1–13 (2018)
Zheng, Z., Wang, P., Liu, W., Li, J., Ye, R., Ren, D.: Distance-IoU loss: faster and better learning for bounding box regression. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 34, pp. 12993–13000 (2020)
Zheng, Z., et al.: Enhancing geometric factors in model learning and inference for object detection and instance segmentation. IEEE Trans. Cybern. 52(8), 8574–8586 (2021)
Acknowledgements
This work is being supported by the National Key Research and Development Project of China under Grant No. 2020AAA0104001, the Zhejiang Lab. under Grant No. 2019KD0AD011005, the Zhejiang Provincial Science and Technology Planning Key Project of China under Grant No. 2021C03129, and the Zhejiang Provincial Natural Science Foundation of China under Grant No.LQ22F020008.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Gao, F., Huang, W., Chen, X., Weng, L. (2024). Traffic Sign Recognition Model Based on Small Object Detection. In: Liu, F., Sadanandan, A.A., Pham, D.N., Mursanto, P., Lukose, D. (eds) PRICAI 2023: Trends in Artificial Intelligence. PRICAI 2023. Lecture Notes in Computer Science(), vol 14327. Springer, Singapore. https://doi.org/10.1007/978-981-99-7025-4_18
Download citation
DOI: https://doi.org/10.1007/978-981-99-7025-4_18
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-7024-7
Online ISBN: 978-981-99-7025-4
eBook Packages: Computer ScienceComputer Science (R0)