skip to main content
10.1145/3591569.3591589acmotherconferencesArticle/Chapter ViewAbstractPublication PagesiciitConference Proceedingsconference-collections
research-article

Traffic Lights Detection and Recognition Method using Deep Learning with Improved YOLOv5 for Autonomous Vehicle in ROS2

Published:13 July 2023Publication History

ABSTRACT

One of the most significant uses of autonomous cars in recent years is the detection of traffic light signals. Deep learning technology, which has a number of benefits including high detection accuracy and quick response to changes, is supporting the development of traffic light recognition under various environmental situations. In this paper, we use two methods to improve the traffic light detection and recognition method. First, we speed up training time by using the K-means clustering algorithm to compress image data. Second, a real time traffic light signal (red, yellow, green) identity based on the You Only Look Once (Yolov5) model is introduced. We utilised a variety of datasets including a freely available Roboflow dataset, a set of data obtained from Gazebo simulator, and a traffic light of CanTho city dataset to train and evaluate the proposed system. Furthermore, our algorithm was validated on a vehicle model in a simulated environment Gazebo of Robot Operating System 2 (ROS2).

References

  1. W.-C. S. Cheng-Chin, Ming-Che Ho, “Detecting and recognizing traffic lights by genetic approximate ellipse detection and spatial texture layouts,” ICIC 2011, 2011.Google ScholarGoogle Scholar
  2. T. H.-P. Tran, C. C. Pham, T. P. Nguyen, T. T. Duong, and J. W. Jeon, “Real-time traffic light detection using color density,” in 2016 IEEE International Conference on Consumer Electronics-Asia (ICCE-Asia). IEEE, pp. 1–4, 2016.Google ScholarGoogle ScholarCross RefCross Ref
  3. J. Levinson, J. Askeland, J. Dolson, and S. Thrun, “Traffic light mapping, localization, and state detection for autonomous vehicles,” Int. Conf. on Robotics and Automation (ICRA), pp. 5784–5791, 2011.Google ScholarGoogle ScholarCross RefCross Ref
  4. Fitri Utaminingrum, Renaldi P. Prasetya, and Rizdania, "Combining Multiple Feature for Robust Traffic Sign Detection," Journal of Image and Graphics, Vol. 8, No. 2, pp. 53-58, June 2020.Google ScholarGoogle ScholarCross RefCross Ref
  5. J. Redmon, S. Divvala, R. Girshick, and A. Farhadi, “You only look once: Unified, real-time object detection,” in 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2016.Google ScholarGoogle ScholarCross RefCross Ref
  6. Ryo Hasegawa, Yutaro Iwamoto, and Yen-Wei Chen, "Robust Japanese Road Sign Detection and Recognition in Complex Scenes Using Convolutional Neural Networks," Journal of Image and Graphics, Vol. 8, No. 3, pp. 59-66, September 2020.Google ScholarGoogle ScholarCross RefCross Ref
  7. W. Liu, D. Anguelov, D. Erhan, C. Szegedy, S. Reed, C.-Y. Fu, and A. C. Berg, “Ssd: Single shot multibox detector,” in European conference on computer vision. Springer, 2016.Google ScholarGoogle ScholarCross RefCross Ref
  8. Z. Gan, L. Wenju, C. Wanghui and S. Pan, "Traffic sign recognition based on improved YOLOv4," 2021 6th International Conference on Intelligent Informatics and Biomedical Sciences (ICIIBMS), 2021.Google ScholarGoogle Scholar
  9. F.-A. Redmon, Joseph, “Yolov3: An incremental improvement,” Tech. Rep., 2018.Google ScholarGoogle Scholar
  10. Cai Z.-X., Gu M.-Q. Accurate Recognition and State Estimation of Traffic Lights in Urban Environment. Control. Decis. 2014.Google ScholarGoogle Scholar
  11. Philipsen M.P., Jensen M.B., Møgelmose A., Moeslund T.B., Trivedi M.M. Traffic Light Detection: A Learning Algorithm and Evaluations on Challenging Dataset; Proceedings of the 2015 IEEE 18th International Conference on Intelligent Transportation Systems; Gran Canaria, Spain. 15–18 September 2015; Piscataway, NJ, USA: IEEE; 2015.Google ScholarGoogle Scholar
  12. Du X.-P., Li Y., Guo Y., Xiong H. Vision-Based Traffic Light Detection for Intelligent Vehicles; Proceedings of the 2017 4th International Conference on Information Science and Control Engineering; Changsha, China. 21–23 July 2017; Piscataway, NJ, USA: IEEE; 2017.Google ScholarGoogle ScholarCross RefCross Ref
  13. Behrendt K., Novak L., Botros R. A Deep Learning Approach to Traffic Lights: Detection, Tracking, and Classification; Proceedings of the 2017 IEEE International Conference on Robotics and Automation; Singapore. 29 May–3 June 2017; Piscataway, NJ, USA: IEEE; 2017.Google ScholarGoogle Scholar
  14. Pan W.-G., Chen Y.-H., Liu B., Shi H.-L. Detection and Recognition of Traffic Lights Based on Faster R-CNN. Transducer Microsyst. Technol. 2019;38:147–149.Google ScholarGoogle Scholar
  15. Pan W.-G., Chen Y.-H., Liu B., Shi H.-L. Detection and Recognition of Traffic Lights Based on Faster R-CNN. Transducer Microsyst. Technol. 2019.Google ScholarGoogle Scholar
  16. Morten B.J., Kamal N., Thomas B. Evaluating State-of-the-art Object Detector on Challenging Traffic Light Data; Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition; Honolulu, HI, USA. 21–26 July 2017.Google ScholarGoogle Scholar
  17. Muller J., Dietmayer K. Detecting Traffic Lights by Single Shot Detection; Proceedings of the 21st IEEE International Conference on Intelligent Transportation Systems; Maui, HI, USA. 4–7 November 2018; Piscataway, NJ, USA: IEEE; 2018.Google ScholarGoogle ScholarDigital LibraryDigital Library
  18. Possatti L.C., Guidolini R., Cardoso V.B., Berriel R.F., Paixão T.M., Badue C., de Souza A.F., Oliveira-Santos T. Traffic Light Recognition Using Deep Learning and Prior Maps for Autonomous Cars; Proceedings of the 2019 International Joint Conference on Neural Networks; Budapest, Hungary. 14–19 July 2019.Google ScholarGoogle ScholarCross RefCross Ref
  19. Du L., Chen W., Fu S., Kong H., Li C., Pei Z. Real-time Detection of Vehicle and Traffic Light for Intelligent and Connected Vehicles Based on YOLOv3 Network; Proceedings of the 2019 5th International Conference on Transportation Information and Safety; Liverpool, UK. 14–17 July 2019.Google ScholarGoogle ScholarCross RefCross Ref
  20. Tran, T. H.-P., & Jeon, J. W. (2020). Accurate Real-Time Traffic Light Detection Using YOLOv4. 2020 IEEE International Conference on Consumer Electronics - Asia (ICCE-Asia). doi:10.1109/icce-asia49877.2020.Google ScholarGoogle ScholarCross RefCross Ref
  21. S. Wang , "K-Means Clustering With Incomplete Data," in IEEE Access, vol. 7, pp. 69162-69171, 2019.Google ScholarGoogle ScholarCross RefCross Ref
  22. Hoang Tran Ngoc and Luyl-Da Quach, “Adaptive Lane Keeping Assist for an Autonomous Vehicle based on Steering Fuzzy-PID Control in ROS” International Journal of Advanced Computer Science and Applications(IJACSA), 13(10), 2022.Google ScholarGoogle Scholar

Index Terms

  1. Traffic Lights Detection and Recognition Method using Deep Learning with Improved YOLOv5 for Autonomous Vehicle in ROS2
        Index terms have been assigned to the content through auto-classification.

        Recommendations

        Comments

        Login options

        Check if you have access through your login credentials or your institution to get full access on this article.

        Sign in
        • Published in

          cover image ACM Other conferences
          ICIIT '23: Proceedings of the 2023 8th International Conference on Intelligent Information Technology
          February 2023
          310 pages
          ISBN:9781450399616
          DOI:10.1145/3591569

          Copyright © 2023 ACM

          Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

          Publisher

          Association for Computing Machinery

          New York, NY, United States

          Publication History

          • Published: 13 July 2023

          Permissions

          Request permissions about this article.

          Request Permissions

          Check for updates

          Qualifiers

          • research-article
          • Research
          • Refereed limited

        PDF Format

        View or Download as a PDF file.

        PDF

        eReader

        View online with eReader.

        eReader

        HTML Format

        View this article in HTML Format .

        View HTML Format