基于反向瓶颈和LCBAM设计的X光违禁品检测  

X-ray Prohibited Items Detection Based on Inverted Bottleneck and Light Convolution Block Attention Module

作　　者：董乙杉 郭靖圆 李明泽 孙嘉傲 卢树华[1,2] DONG Yishan;GUO Jingyuan;LI Mingze;SUN Jiaao;LU Shuhua(College of Information and Cyber Security,People’s Public Security University of China,Beijing 102600,China;Key Laboratory of Security Technology and Risk Assessment Ministry of Public Security,Beijing 102600,China)

机构地区：[1]中国人民公安大学信息网络安全学院,北京102600 [2]公安部安全防范技术与风险评估重点实验室,北京102600

出　　处：《计算机科学与探索》2024年第5期1259-1270,共12页Journal of Frontiers of Computer Science and Technology

基　　金：公安部科技强警基础专项(2021JC03);中央高校基本科研业务费重大项目(21JKF102)。

摘　　要：针对X光违禁品图像姿态与角度变化易漏检误检及困难样本检测准确率低等问题,以YOLOv5网络为基线模型,提出一种融合了反向瓶颈结构和轻量化卷积块注意力模块设计的违禁品检测模型。在主干网络采用反向瓶颈结构设计注重细节特征信息,改进网络应对检测目标大角度变化问题;采用轻量化卷积块注意力机制抑制复杂背景干扰,降低模型参数量;此外,采用高斯误差线性单元激活函数和改进的置信度损失函数增强模型的非线性表达能力,加大对置信度预测的惩罚力度,优化网络对困难样本的检测性能。所提模型在三个大型公开数据集OPIXray、SIXray、HiXray上进行训练和测试,mAP分别达到了91.9%、93.4%和82.2%。结果表明,所提模型能够有效解决基线模型应对X光违禁品角度变化问题,具有较高的检测准确性和稳健性。To resolve the problems of position and angle change causing miss and false detection,low accuracy of difficult samples in X-ray luggage images,using YOLOv5 as the baseline,this paper proposes a model by inverted bottleneck and light convolution block attention module for the X-ray prohibited items detection.The inverted bottleneck design is introduced in the backbone to emphasize the detailed features and improve the model to cope with the large-angle change problem.The light convolution block attention module is used to suppress background interference and reduce model parameter.The Gaussian error linear unit activation function and improved loss function are used to enhance the nonlinear expression ability,increasing the punishment of predicted value to optimize the model's detection ability for difficult samples.The proposed model is trained and tested on three large public datasets OPIXray,SIXray,and HiXray,resulting in the mAP of 91.9%,93.4%,and 82.2%,respectively.The results show that the proposed method can effectively solve the problem of angel change in X-ray luggage,indicating its high accuracy and robustness.

关 键 词：X光图像 违禁品检测 反向瓶颈 轻量化卷积块注意力模块(LCBAM) 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]