[发明专利]一种图像检测方法及装置

权利要求书

1.一种图像检测方法，其特征在于，包括：

获取包含一个或多个待检测物体的图像；

根据神经网络对所述图像进行特征提取，得到N个特征图；所述N为大于或等于2的整数；

对所述N个特征图进行融合处理得到第一特征图H；

对所述第一特征图H进行语义信息提取，得到注意力图；

根据所述注意力图和所述第一特征图H得到第二特征图；

将所述N个特征图依次和所述第二特征图融合，得到检测结果。

2.如权利要求1所述的方法，其特征在于，所述对所述N个特征图进行融合处理得到第一特征图H，包括：

所述N个特征图包括多层；其中，第i+1层的M个特征图是由第i层的N个特征图卷积融合得到,M＝N-1；所述i为大于或等于1的整数；

将最后一层的特征图作为所述第一特征图H。

3.如权利要求2所述的方法，其特征在于，所述第i+1层的M个特征图是由第i层的N个特征图卷积融合得到，包括：

所述M个特征图中的第j个特征图由第i层的第j个特征图和第j+1个特征图融合得到，所述j为大于或等于1的整数。

4.如权利要求1所述的方法，其特征在于，所述对所述第一特征图H进行语义信息提取，得到注意力图，包括：

对所述第一特征图H进行卷积降维，得到特征图Q和特征图K；

根据所述特征图Q和所述特征图K得到注意力权重；

对所述注意力权重进行归一化处理得到所述注意力图。

5.如权利要求1所述的方法，其特征在于，所述对所述第一特征图H进行语义信息提取，得到注意力图之后，还包括:

对所述第一特征图H进行卷积，得到特征图V；

根据所述特征图V和所述第一特征图H得到第三特征图S；其中，所述第三特征图S满足如下公式：

S＝(V*A)+H；

其中，所述S是所述第三特征图，所述V是所述特征图V，所述A是所述注意力图，所述H是所述第一特征图H；

对所述第三特征图S进行卷积生成分割结果；

若所述分割结果与所述注意力图中的标记区域一致，根据所述注意力图和所述第一特征图H得到第二特征图。

6.如权利要求1所述的方法，其特征在于，所述根据所述注意力图和所述第一特征图H得到第二特征图，包括:

所述第二特征图满足如下公式：

F＝H*(A+1)；

其中，所述F是所述第二特征图，所述A是所述注意力图，所述H是所述第一特征图。

7.如权利要求1所述的方法，其特征在于，所述将所述N个特征图依次和所述第二特征图融合，得到检测结果，包括：

将所述第二特征图和所述N个特征图中除最后第一层以外的每层的最后一个特征图融合得到所述检测结果。

8.一种图像检测装置，其特征在于，所述装置包括：

获取模块，用于获取包含一个或多个待检测物体的图像；

处理模块，用于根据神经网络对所述图像进行特征提取，得到N个特征图；所述N为大于或等于2的整数；

所述处理模块，还用于对所述N个特征图进行融合处理得到第一特征图H；

所述处理模块，还用于对所述第一特征图H进行语义信息提取，得到注意力图；

所述处理模块，还用于根据所述注意力图和所述第一特征图H得到第二特征图；

所述处理模块，还用于将所述N个特征图依次和所述第二特征图融合，得到检测结果。

9.一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现权利要求1-7任一项所述方法的步骤。

10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于：所述计算机程序被处理器执行时实现权利要求1-7任一项所述方法的步骤。