• Laser & Optoelectronics Progress
  • Vol. 59, Issue 16, 1610013 (2022)
Hong Zhang and Sicong Zhang*
Author Affiliations
  • School of Automation, Xi’an University of Posts & Telecommunications, Xi’an 710100, Shaanxi , China
  • show less
    DOI: 10.3788/LOP202259.1610013 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    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
    Download Citation