X-Ray Image Detection Algorithm for Prohibited Items Based on Feature Enhancement and Loss Optimization
Downloads
To address the issue of low detection accuracy of prohibited items in X-ray security images caused by varying orientations, different scales, and the intertwining of targets with backgrounds, we propose a novel X-ray image detection algorithm based on feature enhancement and loss optimization. The model is built upon the ConvNext network and incorporates a Directional Channel Attention (DCA) mechanism, which efficiently captures the interaction information of local channels in different directions, thereby enhancing the accuracy of the detection model for prohibited items. Additionally, a multi-scale fusion bypass (MFB) branch is designed after the backbone network to integrate information from feature maps at different layers, thereby mitigating the interference of scale variations on the model. Furthermore, the loss function is redesigned to enable the model to automatically adjust its focus on hard samples, improving the overall detection performance. Experimental results on the SIXray dataset demonstrate that the proposed model achieves a mean Average Precision (mAP) of 91.43%, representing a 9.17% improvement over the original algorithm, thereby validating the effectiveness of the proposed method.
MU Siqi， LIN Jinjian， WANG Haiquan， WEI Xiongzhi. An Algorithm for Detection of Prohibited Items in X-ray Images Based on Improved YOLOv4[J]. Acta Armamentarii, 2021, 42(12): 2675-2683. (in Chinese).
Cheng Lang, Jing Chao, Chen Wenpeng. LLP-NAS: Prohibited Item Detection Algorithm with Neural Network Architecture Search using X-ray Images[J]. Science Technology and Engineering,2024,24(2):665-675. (in Chinese).
Wei, Yanlu, et al. "Occluded prohibited items detection: An x-ray security inspection benchmark andde-occlusion attention module." Proceedings of the28th ACM international conference on multimedia. 2020.
Ke Zhang, Liang Zhang. Multi-Scale Detection for X-Ray Prohibited Items in Complex Background[J]. Laser & Optoelectronics Progress, 2021, 58(22): 2210002. (in Chinese).
Akcay S , Kundegorski M E , Willcocks C G ,et al.Using Deep Convolutional Neural Network Architectures for Object Classification and Detection Within X-Ray Baggage Security Imagery[J].IEEE Transactions on Information Forensics and Security, 2018:2203-2215.DOI:10.1109/TIFS.2018.2812196.
Miao C, Xie L, Wan F, et al. Sixray: A large-scale security inspection x-ray benchmark for prohibited item discovery in overlapping images[C]//Proceedings of the IEEE/CVF conference on computer vision and pattern recognition.2019: 2119-2128.
He, Kaiming, et al. "Deep residual learning for image recognition." Proceedings of the IEEE conference on computer vision and pattern recognition. 2016.
Wei, Yanlu, et al. "Occluded prohibited items detection: An x-ray security inspection benchmark andde-occlusion attention module." Proceedings of the28th ACM international conference on multimedia. 2020.
Shaoqing Yao, Zhigang Su. Prohibited Item Identification Algorithm Based onLightweight Segmentation Network[J]. Laser & Optoelectronics Progress, 2021, 58(2): 0210022. (in Chinese).
FENG X，WEI X K，LIU C H，et al. Contraband classification method for X-ray security images considering sample imbalance[J]. Journal of Beijing University of Aeronautics and Astronautics，2023，49（12）：3215-3221 （in Chinese） doi: 10.13700/j.bh.1001-5965.2022.0095
Yang Cao, Li Zhang, Junxi Meng, Qian Song, Letian Zhang. Multi-Target Prohibited Item Recognition Algorithm for X-Ray Security Scene[J]. Laser & Optoelectronics Progress, 2022, 59(10): 1015009.(in Chinese).
YUAN Jinhao, ZHANG Nanfeng, RUAN Jieshan, GAO Xiangdong. Detection of prohibited items in X-ray images based on modified YOLOX algorithm[J]. Laser Technology, 2023, 47(4): 547. (in Chinese).
Liu, Zhuang, et al. "A convnet for the 2020s." Proceedings of the IEEE/CVF conference on computer vision and pattern recogntion. 2022.
Lin, Tsung-Yi, et al. "Focal loss for dense object detection." Proceedings of the IEEE international conference on computer vision. 2017.
Hu, Jie, Li Shen, and Gang Sun. "Squeeze-and-excitation networks." Proceedings of the IEEE conference on computer vision and pattern recognition. 2018.
Hou Q, Zhou D, Feng J. Coordinate attention for efficient mobile network design[C]//Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2021: 13713-13722.
