Laser & Optoelectronics Progress, Volume. 59, Issue 2, 0210011(2022)

Recognition Algorithm of Dangerous Goods in Security Inspection Based on Multi-Layer Attention Mechanism

Wen Wang1, Yatong Zhou1、*, Baojun Shi2, Hao He1, and Jianwei Zhang1
Author Affiliations
  • 1School of Electronic Information Engineering, Hebei University of Technology, Tianjin 300401, China
  • 2School of Mechanical Engineering, Hebei University of Technology, Tianjin 300401, China
  • show less

    Aiming at the problems that the data set used by the existing recognition algorithms is too simple, the recognition accuracy of dangerous goods in security inspection images in real scenes is low, and it is easy to lead to false detection and missed detection, we propose a class-balanced hierarchical refinement algorithm based on penetration hypothesis, which combines multi-layer channel attention mechanism and space attention mechanism. First, based on the hierarchical modeling of security image, channel attention mechanism is added to the feature map to give different weight to different channel features. Then, spatial attention mechanism is added to give different weight to the unique color features of security image in space. Finally, the residual network is used to add double attention mechanism to different layers of security image for ablation experiment. The experimental results show that after adding double attention mechanism to the fixed two layers at the same time, the network can significantly improve the identification accuracy of dangerous goods in security inspection, and verify the effectiveness and robustness of the multi-layer attention mechanism algorithm.

    Tools

    Get Citation

    Copy Citation Text

    Wen Wang, Yatong Zhou, Baojun Shi, Hao He, Jianwei Zhang. Recognition Algorithm of Dangerous Goods in Security Inspection Based on Multi-Layer Attention Mechanism[J]. Laser & Optoelectronics Progress, 2022, 59(2): 0210011

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Dec. 14, 2020

    Accepted: Mar. 16, 2021

    Published Online: Dec. 23, 2021

    The Author Email: Zhou Yatong (zhouyatong_zw@126.com)

    DOI:10.3788/LOP202259.0210011

    Topics