基于改进YOLOv7的X线图像旋转目标检测

摘要

针对X线图像违禁品目标检测中存在的识别定位困难以及忽略物品方向性的问题,提出了一种基于改进YOLOv7的X线图像旋转目标检测算法。首先,通过在原网络中融合高效注意力机制模块提高模型对深层重要特征的提取能力;然后,改进扩展的高效长程注意力机制的特征融合路径,在模块之间增加跳跃连接和1×1卷积架构,使网络提取更丰富的物品特征;最后,针对X线图像中违禁品放置方向任意的问题,使用密集编码标签表示法对角度进行离散化处理,提高违禁品定位的准确性。实验结果表明,改进的算法在HiXray,OPIXray和PIDray数据集上分别取得了91.2%,92.6%和66.4%的检测精度,较原YOLOv7模型分别提高了20.2%,10.6%和15.5%,在有效提高X线图像违禁品检测精度的基础上,为保障公共安全提供了很好的技术支持。