[发明专利]安检危险品检测方法及系统

权利要求书

1.安检危险品检测方法，用于从安检物品中检测出危险品，其特征在于，包括：

获得安检物品图像；

将所述安检物品图像输入到目标检测算法中，对所述安检物品图像中的危险品图像进行检测，得到检测结果；

根据所述检测结果检测出危险品；

其中，所述目标检测算法的搭建步骤包括：

对输入图像进行特征提取，得到特征向量Z_q，根据特征向量Z_q的上、下、左、右四块区域，提取出输入图像中的四块关注区域；

所述特征向量Z_q通过第一全连接层得到每一块关注区域的向量Q、向量K和向量V；

所述特征向量Z_q通过第二全连接层得到每一块关注区域的注意力权重Att；

根据每一块所述关注区域的注意力权重Att，对每一块关注区域进行自注意力和交叉注意力计算，得到每一块关注区域的重构向量Va；其中，对任一块关注区域Va1的自注意力和交叉注意力计算过程包括：

任一块关注区域的自注意力为：Va1＝Q1×K1×V1；

任一块关注区域和第二关注块区域的交叉注意力为：Va2＝Q1×K2×V1；

任一块关注区域和第三关注块区域的交叉注意力为：Va3＝Q1×K3×V1；

任一块关注区域和第四块关注区域的交叉注意力为：Va4＝Q1×K4×V1；

四块关注区域的所述向量Va通过第三全连接层进行输出，输出结果用于目标检测。

2.根据权利要求1所述的安检危险品检测方法，其特征在于，在所述特征向量Z_q通过第一全连接层之前，还包括：

将所述特征向量Z_q通过第四全连接层得到每一块关注区域的偏移量P；

根据偏移量P对每一块所述关注区域的坐标进行修正，具体为：对任一块关注区域，该关注区域的坐标＝该关注区域的初始化坐标+P，

其中，四块关注区域的初始化坐标分别为：特征向量Z_q上、下、左、右四个区域的坐标。

3.根据权利要求1所述的安检危险品检测方法，其特征在于，在所述目标检测算法的搭建步骤之后还包括：

利用训练集对所述目标检测算法进行训练；

利用测试集对训练后的所述目标检测算法进行测试；

根据测试结果检测所述目标检测算法的性能指标，所述性能指标包括准确率；若所述目标检测算法的准确率低于设定值，再次对所述目标检测算法进行训练，直至所述准确率高于设定值。

所述训练集和所述测试集的获取步骤具体包括：

采集安检物品图像，挑选出含有危险品的图像，且各种类型的危险品图像数量均衡；

对所述危险品图像整理和标注；

将整理和标注的危险品图像进行数据增强处理后，一部分作为训练集，另一部分作为测试集。

4.根据权利要求3所述的安检危险品检测方法，其特征在于，所述准确率通过GIoULoss损失函数获得；

GIoULoss损失函数表示为：

其中，A为预测框的面积，B为标签框的面积，C为包围A和B的最小矩形的面积。