[发明专利]一种基于注意力机制连续视角的危险品检测方法

权利要求书

1.一种基于注意力机制连续视角的危险品检测方法，其特征在于，所述方法包括以下步骤：

步骤1，训练获得危险品分类模型。

步骤2，获取待检测危险品图像，利用训练获得的危险品分类模型，获得检测结果。

2.根据权利要求1所述的检测方法，其特征在于，步骤1包括以下子步骤：

步骤1-1，建立训练集；

步骤1-2，构建分类模型；

步骤1-3，训练分类模型。

3.根据权利要求2所述的检测方法，其特征在于，步骤1-1包括以下子步骤：

步骤1-1-1，获得危险品连续角度下的图像；

步骤1-1-2，对危险品每一角度下的图像赋予权值，获得训练数据集。

4.根据权利要求2所述的检测方法，其特征在于，步骤1-1-1中，采用X射线从不同角度拍摄危险品旋转180°的图像，每隔一定角度拍摄一张；

优选地，危险品每旋转10°，拍摄一张图像。

5.根据权利要求2所述的检测方法，其特征在于，步骤1-2中，所述构建分类模型包括构建注意力模块的步骤，

所述注意力模块包括通道注意力模块和空间注意力模块。

6.根据权利要求5所述的检测方法，其特征在于，所述通道注意力模块采用包括以下步骤的方法判断特征图各通道的权重：

(i)将特征图进行空间信息聚合；

(ii)获得特征的权重系数；

(iii)判断特征图中各个通道的重要程度；

所述空间注意力模块对特征图的位置生成对应的权重，优选按照包括以下步骤的方法进行：

(1)对特征图进行空间信息聚合；

(2)获得特征的权重系数；

(3)获得特征图中位置的权重。

7.根据权利要求1所述的检测方法，其特征在于，步骤2包括以下子步骤：

步骤2-1，获得待检测危险品连续视角的图像；

步骤2-2，利用训练得到的危险品分类模型，得到危险品不同角度的注意力权重；

步骤2-3，将不同角度的图像进行加权求和，获得检测结果。

8.一种计算机可读存储介质，其特征在于，所述介质中存储有基于注意力机制连续视角的危险品检测程序，所述程序被处理器执行时，使得处理器执行权利要求1至7之一所述基于注意力机制连续视角的危险品检测方法的步骤。

9.一种计算机设备，其特征在于，所述设备包括存储器和处理器，所述存储器存储有基于注意力机制连续视角的危险品检测程序，所述程序被处理器执行时，使得处理器执行权利要求1至7之一所述基于注意力机制连续视角的危险品检测方法的步骤。