skip to main content
10.1145/3501409.3501674acmotherconferencesArticle/Chapter ViewAbstractPublication PageseitceConference Proceedingsconference-collections
research-article

UAV Lightweight Object Detection Based on the Improved YOLO Algorithm

Published: 31 December 2021 Publication History

Abstract

Aiming at the characteristics of small objects in low-altitude images, special shooting angles, and variable shooting angles of unmanned aerial vehicles (UAVs), this paper proposes a structural innovation based on the YOLOv5-MobileNetv3Small network model. It transplants the MobileNetv3 network structure and improves the BackBone network structure to solve the problem of inference high-pixel images taking up too much memory for low-power edge computing nodes. The improved YOLO algorithm uses the Visdrone2019-DET dataset to train the network model and uses the Jetson NX edge computing platform on the UAV to process 1920*1080 video streams for testing. The memory usage of the optimized YOLOv5-MobileNetv3Small network model can be reduced by 72.4%.

References

[1]
GIRSHICK R, DONAHUE J, DARRELL T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C]. Proceedings of the IEEE conference on computer vision and pattern recognition. 2014: 580--587.
[2]
GIRSHICK R. Fast r-cnn[C]. Proceedings of the IEEE international conference on computer vision. 2015: 1440--1448.
[3]
REN S, HE K, GIRSHICK R, et al. Faster r-cnn: Towards real-time object detection with region proposal networks[J]. Advances in neural information processing systems, 2015, 28: 91--99.
[4]
LIU W, ANGUELOV D, ERHAN D, et al. SSD: Single shot multibox detector[C]//European conference on computer vision. Springer, Cham, 2016: 21--37.
[5]
REDMON J, FARHADI A. Yolov3: An incremental improvement[J]. arXiv preprint arXiv:1804.02767, 2018.
[6]
REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: Unified, real-time object detection[C]. Proceedings of the IEEE conference on computer vision and pattern recognition. 2016: 779--788.
[7]
BOCHKOVSKIY A, WANG C Y, LIAO H Y M. YOLOv4: Optimal Speed and Accuracy of Object Detection[J]. arXiv preprint arXiv:2004.10934, 2020.
[8]
WANG C Y, LIAO H Y M, WU Y H, et al. CSPNet: A new backbone that can enhance learning capability of CNN[C]. Proceedings of the IEEE/CVF conference on computer vision and pattern recognition workshops. 2020: 390--391.
[9]
LIU S, QI L, QIN H, et al. Path aggregation network for instance segmentation[C]. Proceedings of the IEEE conference on computer vision and pattern recognition. 2018: 8759--8768.
[10]
JOCHER G. YOLOv5[EB/OL]. [2020-08-10]. https://github.com/ultralytics/YOLOv5.
[11]
HOWARD A, SANDLER M, CHU G, et al. Searching for mobilenetv3[C]. Proceedings of the IEEE/CVF International Conference on Computer Vision. 2019: 1314--1324.
[12]
https://github.com/amov-lab/Prometheus.git
[13]
DU D, ZHU P, WEN L, et al. VisDrone-DET2019: The vision meets drone object detection in image challenge results[C]. Proceedings of the IEEE/CVF International Conference on Computer Vision Workshops. 2019: 213--226.

Cited By

View all

Index Terms

  1. UAV Lightweight Object Detection Based on the Improved YOLO Algorithm

    Recommendations

    Comments

    Information & Contributors

    Information

    Published In

    cover image ACM Other conferences
    EITCE '21: Proceedings of the 2021 5th International Conference on Electronic Information Technology and Computer Engineering
    October 2021
    1723 pages
    ISBN:9781450384322
    DOI:10.1145/3501409
    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 ACM 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: 31 December 2021

    Permissions

    Request permissions for this article.

    Check for updates

    Author Tags

    1. UAV
    2. YOLO
    3. edge computing
    4. lightweight
    5. object detection

    Qualifiers

    • Research-article
    • Research
    • Refereed limited

    Conference

    EITCE 2021

    Acceptance Rates

    EITCE '21 Paper Acceptance Rate 294 of 531 submissions, 55%;
    Overall Acceptance Rate 508 of 972 submissions, 52%

    Contributors

    Other Metrics

    Bibliometrics & Citations

    Bibliometrics

    Article Metrics

    • Downloads (Last 12 months)31
    • Downloads (Last 6 weeks)3
    Reflects downloads up to 30 Jan 2025

    Other Metrics

    Citations

    Cited By

    View all

    View Options

    Login options

    View options

    PDF

    View or Download as a PDF file.

    PDF

    eReader

    View online with eReader.

    eReader

    Figures

    Tables

    Media

    Share

    Share

    Share this Publication link

    Share on social media