[发明专利]基于机器学习的安检图像违禁品检测方法、系统及介质

说明书

本申请提供了一种基于机器学习的安检图像违禁品检测方法、系统及介质，获取预设数量的X射线安检图像，形成数据集；数据集包括多种违禁品的X射线安检图像；将数据集输入违禁品分割模型进行模型训练，得到训练后的违禁品分割模型；违禁品分割模型包括编码器网络、全局多尺度模块以及解码器网络；将待检测安检图像输入训练后的违禁品分割模型，得到违禁品检测结果；违禁品检测结果包括违禁品分割结果。本申请通过将基于卷积神经网络的图像分割方法用于X射线安检图像违禁品检测中，实现了获得更准确精细的违禁品分割结果。

技术领域

本申请属于计算机视觉技术领域，具体地，涉及一种基于机器学习的安检图像违禁品检测方法、系统及介质。

背景技术

人工智能对技术的推动提升了社会生产力，推动了生产关系的变革。其中，随着机器学习技术的提高，越来越多的应用于更多领域。

X射线用于行李安全检查是物流行业和公共交通运输非常重要的环节，承担着防止危险品进入货运通道和公共交通给公共安全造成威胁的重要责任。随着物流包裹数量和公共交通旅客吞吐量的剧增，已经远远超过了人工安全检查可以处理的范围，这给人工安检带来了巨大的挑战。因此，利用计算机视觉技术自动识别X射线安检图像中的违禁物品来辅助人工安检缓解人工安检的压力是非常有必要的。

目前，基于计算机视觉的违禁品检测都是基于图像分类和目标检测任务的，但是这些方法只能对X射线安全图像中的违禁品进行粗略的分类和定位，无法精准的确定违禁品的轮廓，并且由于X射线安全图像的透视特性，造成不同类别的违禁物品之前可能存在可见的重叠现象，这与自然图像中的重叠现象是不可见的不一样，容易造成违禁物品的漏检。此外，由于X射线图像是通过X射线扫描仪根据不同材料分配为不同颜色得到，一些安全物品的材料可能和违禁物品的材料很相似，导致在图像中显示颜色差异不大，这样就会造成违禁品边界模糊难以识别的问题。由于在X射线安检图像中，违禁品所占的图像的像素比例比较小，背景所占比例远大于前景，会导致计算损失的时候产生正负样本不均衡的问题。

发明内容

本发明提出的基于机器学习的安检图像违禁品检测方法、装置及介质，可以精准的确定违禁品的轮廓并进行违禁品分割，解决了目前检测存在的违禁物品漏检的问题。

根据本申请实施例的第一个方面，提供了一种基于机器学习的安检图像违禁品检测方法，具体包括以下步骤：

获取预设数量的X射线安检图像，形成数据集；数据集包括多种违禁品的X射线安检图像；

将数据集输入违禁品分割模型进行模型训练，得到训练后的违禁品分割模型；违禁品分割模型包括编码器网络、全局多尺度模块以及解码器网络；

将待检测安检图像输入训练后的违禁品分割模型，得到违禁品检测结果；违禁品检测结果包括违禁品分割结果。

在本申请一些实施方式中，将数据集输入违禁品分割模型进行模型训练，得到训练后的违禁品分割模型，具体包括以下步骤：

将数据集的图像进行违禁品分割形式标注，并通过所述违禁品分割形式标注求解得到违禁品最小外接矩形标注和违禁品边缘标注；

获取数据集中随机划分的训练集图像以及相应标注信息；

对训练集图像进行图像预处理，得到预处理图像；

将预处理图像输入违禁品分割模型的编码器网络、全局多尺度模块以及解码器网络，得到解码特征图；

将解码特征图分别输入三个卷积层，得到违禁品分割输出结果、违禁品最小外接矩形输出结果以及违禁品边缘输出结果；

计算违禁品分割输出结果、违禁品最小外接矩形输出结果以及违禁品边缘输出结果，与违禁品分割形式标注、违禁品最小外接矩形标注和违禁品边缘标注之间的损失，通过随机梯度下降优化算法对模型参数进行优化；