Security Inspection Image Object Detection Method with Attention Mechanism and Multilayer Feature Fusion Strategy

Hong Zhang; Sicong Zhang

doi:10.3788/LOP202259.1610013

[1] Chen Z Q, Zhang L, Jin X. Recent progress on X-ray security inspection technologies[J]. Chinese Science Bulletin, 62, 1350-1365(2017).

[2] Mery D. Computer vision technology for X-ray testing[J]. Insight-Non-Destructive Testing and Condition Monitoring, 56, 147-155(2014).

[3] Jaccard N, Rogers T W, Morton E J et al. Using deep learning on X-ray images to detect threats[D](2016).

[4] Girshick R, Donahue J, Darrell T et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C], 580-587(2014).

[5] He K M, Zhang X Y, Ren S Q et al. Spatial pyramid pooling in deep convolutional networks for visual recognition[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 37, 1904-1916(2015).

[6] Girshick R. Fast R-CNN[C], 1440-1448(2015).

[7] Ren S Q, He K M, Girshick R et al. Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 39, 1137-1149(2017).

[8] Liu W, Anguelov D, Erhan D et al. SSD: single shot MultiBox detector[M]. Leibe B, Matas J, Sebe N, et al. Computer vision-ECCV 2016. Lecture notes in computer science, 9905, 21-37(2016).

[9] Redmon J, Divvala S, Girshick R et al. You only look once: unified, real-time object detection[C], 779-788(2016).

[10] Redmon J, Farhadi A. YOLO9000: better, faster, stronger[C], 6517-6525(2017).

[11] Redmon J, Farhadi A. YOLOv3:an incremental improvement[EB/OL]. https://arxiv.org/abs/1804.02767

[12] Bochkovskiy A, Wang C Y, Liao H Y M. YOLOv4: optimal speed and accuracy of object detection[EB/OL]. https://arxiv.org/abs/2004.10934

[13] Wang C Y, Liao H Y M, Wu Y H et al. CSPNet: a new backbone that can enhance learning capability of CNN[C], 1571-1580(2020).

[14] Li B, Wang C, Wu J et al. Surface defect detection of aeroengine components based on improved YOLOv4 algorithm[J]. Laser & Optoelectronics Progress, 58, 1415004(2021).

[15] Tan S L, Bie X B, Lu G L et al. Real-time detection for mask-wearing of personnel based on YOLOv5 network model[J]. Laser Journal, 42, 147-150(2021).

[16] Liu J N, Nie K. One-stage object detection algorithm based on improved YOLOv3[J]. Electronics Optics & Control, 28, 30-33, 69(2021).

[17] Xu C J, Wang X F, Yang Y D. Attention-YOLO: YOLO detection algorithm that introduces attention mechanism[J]. Computer Engineering and Applications, 55, 13-23, 125(2019).

[18] Li L Y, Liu Q, Zou Y M et al. Rail surface defect detection based on improved YOLOv5 algorithm[J]. Journal of Wuyi University (Natural Science Edition), 35, 43-48, 54(2021).

[19] Zhang Y K, Su Z G, Zhang H G et al. Multi-scale prohibited item detection in X-ray security image[J]. Journal of Signal Processing, 36, 1096-1106(2020).

[20] Zhang Z, Li H F, Li M Z. Research on YOLO algorithm in abnormal security images[J]. Computer Engineering and Applications, 56, 187-193(2020).

[21] Guo S X, Zhang L. YOLO-C: one-stage network for prohibited items detection within X-ray images[J]. Laser & Optoelectronics Progress, 58, 0810003(2021).

[22] Liu Y D, Wang Y T, Wang S W et al. CBNet: a novel composite backbone network architecture for object detection[J]. Proceedings of the AAAI Conference on Artificial Intelligence, 34, 11653-11660(2020).

[23] Liu S, Qi L, Qin H F et al. Path aggregation network for instance segmentation[C], 8759-8768(2018).

[24] Rezatofighi H, Tsoi N, Gwak J et al. Generalized intersection over union: a metric and a loss for bounding box regression[C], 658-666(2019).

[25] Hu J, Shen L, Albanie S et al. Squeeze-and-excitation networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 42, 2011-2023(2020).

[26] Wang Q L, Wu B G, Zhu P F et al. ECA-Net: efficient channel attention for deep convolutional neural networks[C], 11531-11539(2020).

[27] Woo S, Park J, Lee J Y et al. CBAM: convolutional block attention module[EB/OL]. https://arxiv.org/abs/1807.06521

[28] Liang T F, Zhang N F, Zhang Y X et al. Summary of research progress on application of prohibited item detection in X-ray images[J]. Computer Engineering and Applications, 57, 74-82(2021).

[29] Yu F, Koltun V. Multi-scale context aggregation by dilated convolutions[EB/OL]. https://arxiv.org/abs/1511.07122

[30] Liu S T, Huang D, Wang Y H. Receptive field block net for accurate and fast object detection[EB/OL]. https://arxiv.org/abs/1711.07767

[31] Liu S T, Huang D, Wang Y H. Learning spatial fusion for single-shot object detection[EB/OL]. https://arxiv.org/abs/1911.09516