Laser & Optoelectronics Progress, Volume. 59, Issue 16, 1610013(2022)
Security Inspection Image Object Detection Method with Attention Mechanism and Multilayer Feature Fusion Strategy
YOLOv5 (You only look once,v5) is widely used in real-time target recognition because of its fast detection speed and high accuracy. On X-ray security image detection errors or omissions problems with complex backgrounds, multiple scales, and overlapping. By improving the attention mechanism, a new feature fusion strategy is developed based on the YOLOv5s network structure. This study proposes a YOLOv5s-AFA object detection network with an adaptive feature fusion technique and an attention mechanism. In the shallow layer of the network, an extended receptive field module and an improved spatial attention mechanism are introduced, whereas the improved channel attention mechanism is introduced in the deep layer. The new feature fusion technique can output three feature maps of varying depths at a time and fusing shallow spatial and deep semantic information using adaptive learning weights to improve the network learning. The target results on the X-ray security image dataset show that the false and missed detection rates of the YOLOv5s-AFA network decrease considerably compared with other compared networks.
Get Citation
Copy Citation Text
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
Category: Image Processing
Received: Jul. 22, 2021
Accepted: Sep. 24, 2021
Published Online: Jul. 22, 2022
The Author Email: Zhang Sicong (1622065516@qq.com)