Nothing Special   »   [go: up one dir, main page]

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 2024 Publication 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.
[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.
[3]
Glenn J. YOLOv5, 2021, https://github.com/ultralytics/yolov5.
[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.
[5]
Li, Hulin, "Slim-neck by GSConv: A better design paradigm of detector architectures for autonomous vehicles." arXiv preprint arXiv:2206.02424. 2022.
[6]
Wang, Jiaqi, "Carafe: Content-aware reassembly of features." Proceedings of the IEEE/CVF international conference on computer vision. 2019.
[7]
Lin, Tsung-Yi, "Feature pyramid networks for object detection." Proceedings of the IEEE conference on computer vision and pattern recognition. 2017.
[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.
[9]
Ding, Xiaohan, "Repvgg: Making vgg-style convnets great again." Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2021.
[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.
[11]
Everingham, Mark, "The pascal visual object classes challenge: A retrospective." International journal of computer vision 111, 2015: 98-136.
[12]
Geiger, Andreas, "Vision meets robotics: The kitti dataset." The International Journal of Robotics Research 32.11, 2013: 1231-1237.
[13]
Han, Jianhua, "Soda10m: Towards large-scale object detection benchmark for autonomous driving." 2023.
[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.
[15]
Glenn J, Ayush C, Jing Q. YOLO by Ultralytics, 2023, https://github.com/ultralytics/ultralytics.

Index Terms

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

    Recommendations

    Comments

    Please enable JavaScript to view thecomments powered by Disqus.

    Information & Contributors

    Information

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

    Permissions

    Request permissions for this article.

    Check for updates

    Qualifiers

    • Research-article
    • Research
    • Refereed limited

    Conference

    SPCNC 2023

    Contributors

    Other Metrics

    Bibliometrics & Citations

    Bibliometrics

    Article Metrics

    • 0
      Total Citations
    • 20
      Total Downloads
    • Downloads (Last 12 months)20
    • Downloads (Last 6 weeks)5
    Reflects downloads up to 22 Nov 2024

    Other Metrics

    Citations

    View Options

    Login options

    View options

    PDF

    View or Download as a PDF file.

    PDF

    eReader

    View online with eReader.

    eReader

    HTML Format

    View this article in HTML Format.

    HTML Format

    Media

    Figures

    Other

    Tables

    Share

    Share

    Share this Publication link

    Share on social media