Laser & Optoelectronics Progress, Volume. 61, Issue 24, 2437010(2024)

High-Precision and Lightweight Object Detection Model for Drone Aerial Photography Images

Xiangyang Zhao1, Zaifeng Shi1,3、*, Yunfeng Wang1, Xiaowei Niu1, and Tao Luo2
Author Affiliations
  • 1School of Microelectronics, Tianjin University, Tianjin 300072, China
  • 2College of Intelligence and Computing, Tianjin University, Tianjin 300072, China
  • 3Tianjin Key Laboratory of Imaging and Sensing Microelectronic Technology, Tianjin 300072, China
  • show less

    The current mainstream lightweight object detection models exhibit low detection accuracy in unmanned aerial vehicle (UAV) photography scenes. This study introduces a high-precision and lightweight aerial photography image object detection model based on YOLOv8s, named LEFE-YOLOv8. First, an enhanced feature extraction convolution (EFEConv) incorporating an attention mechanism was developed. It is integrated with partial channel convolution (PConv) and 1×1 convolution to create a lightweight enhanced feature extraction module. This integration augments the model's feature extraction capabilities and reduces the number of parameters and computational complexity. Subsequently, a lightweight dynamic upsampling operator module was incorporated into the feature fusion network, effectively addressing the information loss problem during the upsampling process in high-level feature networks. Finally, a detection head with multi-scale modules was designed to enhance the network model's multi-scale detection capabilities. The final experimental results demonstrate that, compared with the benchmark model, the improved model achieves an average accuracy of 42.3% and 83.9% on the VisDrone2019 and HIT-UAV datasets, respectively, with less than 10×106 parameters. These results establish the model's suitability for aerial image object detection tasks.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Xiangyang Zhao, Zaifeng Shi, Yunfeng Wang, Xiaowei Niu, Tao Luo. High-Precision and Lightweight Object Detection Model for Drone Aerial Photography Images[J]. Laser & Optoelectronics Progress, 2024, 61(24): 2437010

    Download Citation

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

    Category: Digital Image Processing

    Received: Apr. 15, 2024

    Accepted: May. 20, 2024

    Published Online: Dec. 16, 2024

    The Author Email: Zaifeng Shi (shizaifeng@tju.edu.cn)

    DOI:10.3788/LOP241103

    CSTR:32186.14.LOP241103

    Topics