Hong Zhang, Sicong Zhang. Security Inspection Image Object Detection Method with Attention Mechanism and Multilayer Feature Fusion Strategy[J]. Laser & Optoelectronics Progress, 2022, 59(16): 1610013
- Laser & Optoelectronics Progress
- Vol. 59, Issue 16, 1610013 (2022)
Fig. 1. Structure of the YOLOv5s network
Fig. 2. Structure of the SE module
Fig. 3. Structure of the SA module
Fig. 4. Structure of the YOLOv5s-AFA network
Fig. 5. Structure of the ERF module
Fig. 6. Structure of the iECA module
Fig. 7. Attention structure of the ASFF module
Fig. 8. X-ray security image dataset. (a)‒(f) Image 1‒ image 6
Fig. 9. Label distribution of training set. (a) Center point distribution; (b) width and height distribution
Fig. 10. Loss and mAP of five networks. (a) Loss curve; (b) mAP curve
Fig. 11. Comparison of detection results for X-ray images. (a) YOLOv4; (b) PP-YOLOv2; (c) YOLOv5x; (d) YOLOv5s; (e) YOLOv5s-AFA
|
Table 1. Number of convolution kernels of Backbone in different YOLOv5 networks
|
Table 2. Comparison results of SA modules
|
Table 3. Comparison results of channel attention module
|
Table 4. Comparison results of ablation experiments of each module
|
Table 5. Detection results of different networks
Set citation alerts for the article
Please enter your email address
© Copyright 2018-2021 | Chinese Laser Press. All Rights Reserved 沪ICP备15018463号-20