Laser & Optoelectronics Progress, Volume. 60, Issue 10, 1011002(2023)
Infrared Target Detection Method Based on Attention Mechanism
Figures & Tables(14)
Fig. 7. Backbone network and decoder network. (a) Backbone network; (b) decoder network
Fig. 8. Graphs of detection results. (a) YOLO v3; (b) E-YOLO; (c) scSE-IYOLOv4; (d) proposed algorithm
Table 1. Backbone network structure
View tableTable 1. Backbone network structure
Input Operator t c n s 2242×3 Conv2D3×3 32 1 2 1122×32 Pbottleneck 1 16 1 1 1122×16 Pbottleneck 6 24 2 2 562×24 Pbottleneck 6 32 3 2 282×32 Pbottleneck 6 64 4 1 142×64 Pbottleneck 6 96 3 2 142×96 Pbottleneck 6 160 3 2 72×160 Pbottleneck 6 320 1 1 72×320 Conv2D1×1 1280 1 1 72×1280 AvgPool7×7 1 1×1×1280 Conv2D1×1 1280
Table 2. Experimental results of four algorithms on FLIR dataset
View tableTable 2. Experimental results of four algorithms on FLIR dataset
Algorithm Model size /MB mAP /%（IoU∶0.5） FPS YOLOv3 89.4 75.6 54.7 E-YOLO 85.3 77.3 55.6 scSE-IYOLOv4 93.1 79.3 53.9 Proposed algorithm 80.5 81.4 56.6
Table 3. Experimental results of four algorithms on TNO dataset
View tableTable 3. Experimental results of four algorithms on TNO dataset
Algorithm Model size /MB mAP /%（IoU∶0.5） FPS YOLOv3 89.4 84.2 60.5 E-YOLO 85.3 85.7 62.3 scSE-IYOLOv4 93.1 87.4 61.7 Proposed algorithm 80.5 88.1 62.8
Table 4. Comparison of results of ablation experiment 1
View tableTable 4. Comparison of results of ablation experiment 1
Experiment grouping Only-channel Only-spatial CBAM PAM mAP /% FPS group 1 √ × × × 77.9 57.4 group 2 × √ × × 78.5 56.8 group 3 × × √ × 80.6 56.1 group 4 × × × √ 81.4 56.5
Table 5. Comparison of results of ablation experiment 2
View tableTable 5. Comparison of results of ablation experiment 2
Experiment grouping FPN RFP T-FPN mAP /% FPS group 1 √ × × 77.6 57.9 group 2 × √ × 81.4 56.6 group 3 × × √ 82.5 52.8
Table 6. Comparison of results of ablation experiment 3
View tableTable 6. Comparison of results of ablation experiment 3
Experiment grouping Loss CIoU GHM mAP /% FPS group 1 √ × × 77.9 55.7 group 2 × √ × 78.2 56.2 group 3 × × √ 78.5 56.1 group 4 × √ √ 81.4 56.6
Tools
Get Citation
Copy Citation Text
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
Paper Information
Category: Imaging Systems
Received: Dec. 20, 2021
Accepted: Feb. 14, 2022
Published Online: May. 23, 2023
The Author Email: Weida Zhan (zhanweida@cust.edu.cn)
© Copyright 2018-2021 | Chinese Laser Press.
All Rights Reserved 沪ICP备15018463号-20