Optics and Precision Engineering, Volume. 31, Issue 18, 2723(2023)

Joint self-attention and branch sampling for object detection on drone imagery

Yunzuo ZHANG1...2,*, Cunyu WU1, Yameng LIU1, Tian ZHANG1 and Yuxin ZHENG1 |Show fewer author(s)
Author Affiliations
  • 1School of Information Science and Technology, Shijiazhuang Tiedao University, Shijiazhuang050043, China
  • 2Hebei Key Laboratory of Electromagnetic Environmental Effects and Information Processing, Shijiazhuang Tiedao University, Shijiazhuang050043, China
  • show less

    Object detection on drone imagery is widely used in many fields. However, due to the complexity of the image background, the dense small objects and the dramatic scale changes, the existing object detection on drone imagery methods are not accurate enough. In order to solve this problem, we propose an accurate object detection method for drone imagery joint self attention and branch sampling. Firstly, a nested residual structure integrating self attention and convolution is designed to achieve the effective combination of global and local information, which makes the model to focus on the object area and ignore invalid features. Secondly, we design a feature fusion module based on branch sampling to mitigate the loss of object information. Finally, an improved detector for small objects is added to alleviate the problem of sharp scale changes. Furthermore, we propose a feature enhancement module to obtain more discriminative small object features. The experimental results show that the proposed algorithm performs well in various scenarios. Specifically, the mAP50 and mAP of the s model on the VisDrone2019 reached 59.3% and 37.1% respectively, an increase of 5.6% and 5.4% compared with the baseline. The mAP50 and mAP on the UAVDT reached 44.1% and 24.9% respectively, an increase of 5.8% and 3.2% compared with the baseline.

    Tools

    Get Citation

    Copy Citation Text

    Yunzuo ZHANG, Cunyu WU, Yameng LIU, Tian ZHANG, Yuxin ZHENG. Joint self-attention and branch sampling for object detection on drone imagery[J]. Optics and Precision Engineering, 2023, 31(18): 2723

    Download Citation

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

    Category: Information Sciences

    Received: Dec. 29, 2022

    Accepted: --

    Published Online: Oct. 12, 2023

    The Author Email: ZHANG Yunzuo (zhangyunzuo888@sina.com)

    DOI:10.37188/OPE.20233118.2723

    Topics