计算机科学与探索

作者：董乙杉, 郭靖圆, 李明泽, 孙嘉傲, 卢树华

单位：1. 中国人民公安大学 信息网络安全学院，北京 102600 2. 公安部安全防范技术与风险评估重点实验室，北京 102600

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

针对X光违禁品图像姿态与角度变化易漏检误检及困难样本检测准确率低等问题，以YOLOv5网络为基线模型，提出一种融合了反向瓶颈结构和轻量化卷积块注意力模块设计的违禁品检测模型。在主干网络采用反向瓶颈结构设计注重细节特征信息，改进网络应对检测目标大角度变化问题；采用轻量化卷积块注意力机制抑制复杂背景干扰，降低模型参数量；此外，采用高斯误差线性单元激活函数和改进的置信度损失函数增强模型的非线性表达能力，加大对置信度预测的惩罚力度，优化网络对困难样本的检测性能。所提模型在三个大型公开数据集OPIXray、SIXray、HiXray上进行训练和测试，mAP分别达到了91.9%、93.4%和82.2%。结果表明，所提模型能够有效解决基线模型应对X光违禁品角度变化问题，具有较高的检测准确性和稳健性。

来源：2024年第5期

《计算机科学与探索》期刊编辑部

查看计算机科学与探索杂志2024年第5期