Optics and Precision Engineering, Volume. 31, Issue 20, 3021(2023)

Lightweight target detection network for UAV platforms

Dandan HUANG1... Han GAO1, Zhi LIU1,2,*, Lintao YU1 and Huiji WANG1 |Show fewer author(s)
Author Affiliations
  • 1School of Electronics and In formation Engineering, Changchun University of Science and Technology, Changchun30022, China
  • 2National and Local Joint Engineering Research Center of Space Photoelectric Technology, Changchun University of Science and Technology, Changchun1300, China
  • show less

    A lightweight target detection network for application to unmanned aerial vehicle (UAV) platforms was proposed for solving the problems of large image-scale variation, small target size, and limited embedded computing resources on UAVs in UAV-side target detection. The network used YOLOv5 as the benchmark model. First, detection branches were used to solve the problem of scale variation. Then, a small-target detection metric based on a mixture of normalized Wasserstein distance and traditional IOU was used for solving the problem of inaccurate small-target detection. In addition, a C3_FN lightweight network structure combining FasterNet and C3 was employed to reduce the computational burden of the network and make it more suitable for UAV platforms. The performance of the algorithms was tested on a simulation platform and an embedded platform using the UAV target detection dataset VisDrone. The simulation platform test results indicate that the proposed network achieves improvements of 6.6% and 4.8% in the mAP0.5 and mAP0.5-0.95 metrics, respectively, compared with a benchmark network, and the inference time is only 45.9 ms. The detection results are superior to those of mainstream UAV target detection networks. The test results for the embedded device (NVIDIA Jetson Nano) indicate that the proposed algorithm can achieve high accuracy and near real-time detection performance with limited hardware resources.

    Tools

    Get Citation

    Copy Citation Text

    Dandan HUANG, Han GAO, Zhi LIU, Lintao YU, Huiji WANG. Lightweight target detection network for UAV platforms[J]. Optics and Precision Engineering, 2023, 31(20): 3021

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Information Sciences

    Received: Apr. 24, 2023

    Accepted: --

    Published Online: Nov. 28, 2023

    The Author Email: LIU Zhi (liuzhi@cust.edu.cn)

    DOI:10.37188/OPE.20233120.3021

    Topics