基于可变形卷积与注意力机制的X光安检违禁品检测
DOI:
作者:
作者单位:

西华大学计算机与软件工程学院 成都 610039

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

四川省自然科学基金(2022NSFSC0533)、西华大学研究生创新基金(ycjj2021030)项目资助


Prohibited items detection based on deformable convolution and attention mechanism in X-ray security inspection
Author:
Affiliation:

School of Computer and Software Engineer, Xihua University,Chengdu 610039, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    X光安检违禁品检测被广泛应用于维护公共交通安全和人身安全。针对X光图像中违禁品存在形状尺度多变、重叠遮挡严重等问题,提出一种结合可变形卷积与注意力机制改进的YOLOv5s模型用于违禁品检测。首先在主干网络中引入可变形卷积,通过学习采样偏移量来适应物体的不同形变,增强空间特征信息提取能力;其次利用混合卷积注意力模块加强模型对检测目标的感知能力,抑制无关背景干扰;然后构造通道引导的空洞空间金字塔模块,获取更加准确的全局上下文信息,提高模型对重叠遮挡目标的识别能力;最后采用CARAFE算子代替最近邻插值,在上采样过程中充分利用内容信息,提高模型检测精度。在SIXray_OD和OPIXray数据集上实验结果显示,所提出模型的mAP@05相较于原YOLOv5s分别提高了21%和18%,达到了906%和900%。与现有诸多先进算法相比,具有较好的检测精度与实时性。

    Abstract:

    Prohibited items detection in X-ray security inspection is widely used to maintain public traffic safety and personal safety. In order to solve the problems of variable shape and scale, severe overlap and occlusion in Xray images, an improved YOLOv5s model combining deformable convolution and attention mechanism is proposed for prohibited items detection. Firstly, deformable convolution is introduced into the backbone network to enhance spatial feature information extraction by learning sampling offsets to adapt to different deformations of objects. Secondly, the mixed convolution attention module is used to enhance the model’s ability to perceive the detected target and suppress irrelevant background interference. Then a channelguided atrous space pyramid module is constructed to obtain more accurate global contextual information and improve the model′s ability to identify overlapping occlusion targets. Finally, the CARAFE operator is used to replace the nearest neighbor interpolation to make full use of the content information in the upsampling process and improve model’s detection accuracy. The experimental results on the SIXray_OD and OPIXray datasets show that the model’s mAP@05 is 21% and 18% higher than the original YOLOv5s, reaching 906% and 900%, respectively. Compared with many existing advanced algorithms, it has better detection accuracy and realtime performance.

    参考文献
    相似文献
    引证文献
引用本文

粟兴旺,王晓明,黄金玻,许茹玉,吴琳.基于可变形卷积与注意力机制的X光安检违禁品检测[J].电子测量技术,2023,46(10):98-108

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-01-09
  • 出版日期: