• Laser & Optoelectronics Progress
  • Vol. 60, Issue 10, 1011002 (2023)
Xing Gu, Weida Zhan*, Ziwei Cui, Tingting Gui..., Yanli Shi and Jiahui Hu|Show fewer author(s)
Author Affiliations
  • School of Electronic Information Engineering, Changchun University of Science and Technology, Changchun 130022, Jilin, China
  • show less
    DOI: 10.3788/LOP213287 Cite this Article Set citation alerts
    Xing Gu, Weida Zhan, Ziwei Cui, Tingting Gui, Yanli Shi, Jiahui Hu. Infrared Target Detection Method Based on Attention Mechanism[J]. Laser & Optoelectronics Progress, 2023, 60(10): 1011002 Copy Citation Text show less
    PAM module
    Fig. 1. PAM module
    Channel attention module
    Fig. 2. Channel attention module
    Spatial attention module
    Fig. 3. Spatial attention module
    Pbottleneck module
    Fig. 4. Pbottleneck module
    Overall network framework of proposed algorithm
    Fig. 5. Overall network framework of proposed algorithm
    Multiscale detection network architecture
    Fig. 6. Multiscale detection network architecture
    Backbone network and decoder network. (a) Backbone network; (b) decoder network
    Fig. 7. Backbone network and decoder network. (a) Backbone network; (b) decoder network
    Graphs of detection results. (a) YOLO v3; (b) E-YOLO; (c) scSE-IYOLOv4; (d) proposed algorithm
    Fig. 8. Graphs of detection results. (a) YOLO v3; (b) E-YOLO; (c) scSE-IYOLOv4; (d) proposed algorithm
    InputOperatortcns
    2242×3Conv2D3×33212
    1122×32Pbottleneck11611
    1122×16Pbottleneck62422
    562×24Pbottleneck63232
    282×32Pbottleneck66441
    142×64Pbottleneck69632
    142×96Pbottleneck616032
    72×160Pbottleneck632011
    72×320Conv2D1×1128011
    72×1280AvgPool7×71
    1×1×1280Conv2D1×11280
    Table 1. Backbone network structure
    AlgorithmModel size /MBmAP /%(IoU∶0.5)FPS
    YOLOv389.475.654.7
    E-YOLO85.377.355.6
    scSE-IYOLOv493.179.353.9
    Proposed algorithm80.581.456.6
    Table 2. Experimental results of four algorithms on FLIR dataset
    AlgorithmModel size /MBmAP /%(IoU∶0.5)FPS
    YOLOv389.484.260.5
    E-YOLO85.385.762.3
    scSE-IYOLOv493.187.461.7
    Proposed algorithm80.588.162.8
    Table 3. Experimental results of four algorithms on TNO dataset
    Experiment groupingOnly-channelOnly-spatialCBAMPAMmAP /%FPS
    group 1×××77.957.4
    group 2×××78.556.8
    group 3×××80.656.1
    group 4×××81.456.5
    Table 4. Comparison of results of ablation experiment 1
    Experiment groupingFPNRFPT-FPNmAP /%FPS
    group 1××77.657.9
    group 2××81.456.6
    group 3××82.552.8
    Table 5. Comparison of results of ablation experiment 2
    Experiment groupingLossCIoUGHMmAP /%FPS
    group 1××77.955.7
    group 2××78.256.2
    group 3××78.556.1
    group 4×81.456.6
    Table 6. Comparison of results of ablation experiment 3
    Xing Gu, Weida Zhan, Ziwei Cui, Tingting Gui, Yanli Shi, Jiahui Hu. Infrared Target Detection Method Based on Attention Mechanism[J]. Laser & Optoelectronics Progress, 2023, 60(10): 1011002
    Download Citation