[发明专利]一种基于多尺度注意力与数据增强的安检违禁品检测方法

摘要

本发明公开了一种基于多尺度注意力与数据增强的安检违禁品检测方法。所述方法包括以下步骤：构建违禁品目标检测模型，包括骨干网络；在自然图像分类数据集上对骨干网络进行训练，得到骨干网络的预训练权重，然后在自然图像目标检测数据集上对加载骨干网络的预训练权重后的违禁品目标检测模型进行训练，得到违禁品目标检测模型的预训练权重；初始化违禁品目标检测模型，在安检X光图像数据集上重新训练，得到训练好的违禁品目标检测模型；对训练好的违禁品目标检测模型进行调整，输入待检测X光图像，得到违禁品的类别和边界框。本发明能够在背景杂乱、物品相互重叠的X光图像中有效地检测出违禁品目标的位置，提高安检的效率和可靠性。