Infrared and Laser Engineering, Volume. 51, Issue 9, 20211101(2022)

Image recognition method of anti drone system based on coordinate attention mechanism

Shan Xue1,2, Yuchao Chen1, Qiongying Lv1、*, and Guohua Cao2
Author Affiliations
  • 1College of Mechanical and Electrical Engineering, Changchun University of Science and Technology, Changchun 130022, China
  • 2Chongqing Research Institute, Changchun University of Science and Technology, Chongqing 400000, China
  • show less
    Figures & Tables(17)
    Effect drawing of image preprocessing
    Images display of drone image dataset
    Diagram sketch of Dense Block
    Structure diagram of TL-DenseNet-121
    Different attention mechanism modules. (a) SE Block; (b) CA Block
    Dense convolutional network incorporating different number of SE Blocks in different locations. (a) TL-SE1-DenseNet-121; (b) TL-SE4-DenseNet-121
    Dense convolutional network incorporating different number of CA Blocks in different locations. (a) TL-CA1-DenseNet-121; (b) TL-CA4-DenseNet-121
    Experimental scene pictures
    Initial training results of TL-DenseNet-121
    Complete training results of TL-DenseNet-121
    Training results of DenseNet-121
    Complete training results of TL-CA4-DenseNet-121
    Comparison curves of various networks. (a) Network comparison curves introduced by SE Block; (b) Network comparison curves introduced by CA Block
    Comparison curves of various models
    • Table 1. Distribution of drone image dataset

      View table
      View in Article

      Table 1. Distribution of drone image dataset

      Simple backgroundComplex background
      DJI mavic 21256960
      DJI mavic air1088960
      Homemade Quad-Rotor drone976624
      Caltech-UCSD-Birds-200-20112400
      Total8264
    • Table 2. Comparison results of different networks

      View table
      View in Article

      Table 2. Comparison results of different networks

      Average accuracyF1-ScoreTraining time/s
      TL-DenseNet-12195.24%0.93454985
      TL-SE1-DenseNet-12195.85%0.93754984
      TL-SE4-DenseNet-12196.84%0.96866362
      TL-CA1-DenseNet-12195.70%0.94905602
      TL-CA4-DenseNet-12197.93%0.98266832
    • Table 3. Comparison results of different models

      View table
      View in Article

      Table 3. Comparison results of different models

      Average accuracyF1-ScoreTraining time/s
      TL-AlexNet94.06%0.91511219
      TL-VGG-1694.49%0.92446002
      TL-ResNet-15295.42%0.93459023
      TL-EfficientNet-B092.59%0.88533388
      TL-CA4-DenseNet-12197.93%0.98266832
    Tools

    Get Citation

    Copy Citation Text

    Shan Xue, Yuchao Chen, Qiongying Lv, Guohua Cao. Image recognition method of anti drone system based on coordinate attention mechanism[J]. Infrared and Laser Engineering, 2022, 51(9): 20211101

    Download Citation

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

    Category: Image processing

    Received: Dec. 20, 2021

    Accepted: --

    Published Online: Jan. 6, 2023

    The Author Email:

    DOI:10.3788/IRLA20211101

    Topics