skip to main content
10.1145/3654446.3654464acmotherconferencesArticle/Chapter ViewAbstractPublication PagesspcncConference Proceedingsconference-collections
research-article

YOLOGP: A YOLOv5-Based Lightweight Network for Efficient Vehicle Detection in Autonomous Driving Scenarios

Published:03 May 2024Publication History

ABSTRACT

The object detection technology holds paramount significance in realizing autonomous driving and AI-assisted driving systems. Swift and precise object detection is crucial for enhancing the safety of autonomous vehicles. However, for in-vehicle edge computing platforms, colossal models fall short of meeting real-time detection requirements, while lightweight models often compromise on detection accuracy. Addressing this issue, this paper proposes an improved real-time object detection algorithm based on YOLOv5.In the proposed method, we combine the One-Shot Aggregation (OSA) concept with the progressive channel compression idea and introduce GSConv to innovatively propose the GSPCA structure. It aims to improve some of the problems exposed by the original C3 structure, so as to enhance the model efficiency. Secondly, we also apply GSConv to the neck network of YOLOv5 and introduce the Content-Aware ReAssembly of Features (CARAFE) upsampling operator in the FPN structure, which utilizes its spatial perception and large receptive field to improve the quality of the upsampling, thus enhancing the feature fusion performance of the network. Experimental results demonstrate that, compared to the baseline, our proposed model achieves the highest improvement of 5.2% in [email protected]:0.95 on the PASCAL VOC dataset, KITTI dataset, and SODA10m dataset. Furthermore, the model's parameter count and computational load are slightly less than those of the original model.

References

  1. Liu, Shaoshan, "Edge computing for autonomous driving: Opportunities and challenges." Proceedings of the IEEE 107.8. 2019: 1697-1716.Google ScholarGoogle Scholar
  2. Jabir, Brahim, Noureddine Falih, and Khalid Rahmani. "Accuracy and efficiency comparison of object detection open-source models." International Journal of Online & Biomedical Engineering 17.5. 2021.Google ScholarGoogle Scholar
  3. Glenn J. YOLOv5, 2021, https://github.com/ultralytics/yolov5.Google ScholarGoogle Scholar
  4. Lee, Youngwan, "An energy and GPU-computation efficient backbone network for real-time object detection." Proceedings of the IEEE/CVF conference on computer vision and pattern recognition workshops. 2019.Google ScholarGoogle Scholar
  5. Li, Hulin, "Slim-neck by GSConv: A better design paradigm of detector architectures for autonomous vehicles." arXiv preprint arXiv:2206.02424. 2022.Google ScholarGoogle Scholar
  6. Wang, Jiaqi, "Carafe: Content-aware reassembly of features." Proceedings of the IEEE/CVF international conference on computer vision. 2019.Google ScholarGoogle Scholar
  7. Lin, Tsung-Yi, "Feature pyramid networks for object detection." Proceedings of the IEEE conference on computer vision and pattern recognition. 2017.Google ScholarGoogle Scholar
  8. Wang, Chien-Yao, "CSPNet: A new backbone that can enhance learning capability of CNN." Proceedings of the IEEE/CVF conference on computer vision and pattern recognition workshops. 2020.Google ScholarGoogle Scholar
  9. Ding, Xiaohan, "Repvgg: Making vgg-style convnets great again." Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2021.Google ScholarGoogle Scholar
  10. Wang, Chien-Yao, Hong-Yuan Mark Liao, and I-Hau Yeh. "Designing network design strategies through gradient path analysis." arXiv preprint arXiv:2211.04800. 2022.Google ScholarGoogle Scholar
  11. Everingham, Mark, "The pascal visual object classes challenge: A retrospective." International journal of computer vision 111, 2015: 98-136.Google ScholarGoogle Scholar
  12. Geiger, Andreas, "Vision meets robotics: The kitti dataset." The International Journal of Robotics Research 32.11, 2013: 1231-1237.Google ScholarGoogle Scholar
  13. Han, Jianhua, "Soda10m: Towards large-scale object detection benchmark for autonomous driving." 2023.Google ScholarGoogle Scholar
  14. Wang, Chien-Yao, Alexey Bochkovskiy, and Hong-Yuan Mark Liao. "YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors." Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2023.Google ScholarGoogle Scholar
  15. Glenn J, Ayush C, Jing Q. YOLO by Ultralytics, 2023, https://github.com/ultralytics/ultralytics.Google ScholarGoogle Scholar

Index Terms

  1. YOLOGP: A YOLOv5-Based Lightweight Network for Efficient Vehicle Detection in Autonomous Driving Scenarios

    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
      SPCNC '23: Proceedings of the 2nd International Conference on Signal Processing, Computer Networks and Communications
      December 2023
      435 pages
      ISBN:9798400716430
      DOI:10.1145/3654446

      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: 3 May 2024

      Permissions

      Request permissions about this article.

      Request Permissions

      Check for updates

      Qualifiers

      • research-article
      • Research
      • Refereed limited
    • Article Metrics

      • Downloads (Last 12 months)4
      • Downloads (Last 6 weeks)4

      Other Metrics

    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