[发明专利]一种目标检测方法、装置、存储介质及电子设备

权利要求书

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

获取待检测图像；

将所述待检测图像输入到预先训练的目标检测模型中，得到所述目标检测模型输出的目标检测图像，其中，所述目标检测模型包括特征提取模块、空间注意模块和通道注意模块，所述特征提取模块中用于提取不同层次的特征信息，所述空间注意模块用于对所述特征提取模块提取的第一特征信息进行空间特征加权，所述通道注意模块用于对所述特征提取模块提取的第二特征信息进行通道加权，经过空间特征加权后的第一特征信息和通道加权后的第二特征信息通过图像融合得到目标检测图像。

2.根据权利要求1所述的方法，其特征在于，在获取待检测图像之后，所述方法还包括：

确定所述待检测图像的先验检测图像；

相应的，所述将所述待检测图像输入到预先训练的目标检测模型中，得到所述目标检测模型输出的检测结果，包括：

将所述待检测图像和所述先验检测图像输入到预先训练的目标检测模型中，得到所述目标检测模型输出的检测结果。

3.根据权利要求1所述的方法，其特征在于，所述空间注意模块用于对第一特征信息分别进行平均池化处理和最大池化处理，将平均池化处理结果和最大化处理结果拼接后进行卷积处理，将卷积处理结果进行非线性处理，得到空间特征信息，将所述空间特征信息加权到所述第一特征信息上。

4.根据权利要求1所述的方法，其特征在于，所述通道注意模块用于对第二特征信息进行平均池化处理，并将平均池化处理结果依次输入至两个全连接层，并经归一化处理后得到通道权重，基于通道权重对所述第二特征信息中的各通道特征进行加权处理。

5.根据权利要求1-4任一所述的方法，其特征在于，所述特征提取模块包括用于提取浅层特征信息的第一特征提取子模块和用于提取深层特征信息的第二特征提取子模块，所述第一特征提取子模块与所述空间注意模块连接，所述第二特征提取子模块与所述通道注意模块连接。

6.根据权利要求5所述的方法，其特征在于，所述第一特征提取子模块中包括至少一个卷积块，每一卷积块分别输出第一特征信息；所述第二特征提取子模块中包括至少一个卷积块，用于每一卷积块分别输出第二特征信息。

7.根据权利6所述的方法，其特征在于，所述目标检测模型还包括：

第一上采样模块，设置在所述第一特征提取子模块与所述空间注意模块之间，用于对所述第一特征信息上采样到第一预设尺寸；

第二上采样模块，设置在所述第二特征提取子模块与所述通道注意模块之间，用于对所述第二特征信息上采样至第二预设尺寸。

8.一种目标检测装置，其特征在于，包括：

图像获取模块，用于获取待检测图像；

目标检测模块，用于将所述待检测图像输入到预先训练的目标检测模型中，得到所述目标检测模型输出的目标检测图像，其中，所述目标检测模型包括特征提取模块、空间注意模块和通道注意模块，所述特征提取模块中用于提取不同层次的特征信息，所述空间注意模块用于对所述特征提取模块提取的第一特征信息进行空间特征加权，所述通道注意模块用于对所述特征提取模块提取的第二特征信息进行通道加权，经过空间特征加权后的第一特征信息和通道加权后的第二特征信息通过图像融合得到目标检测图像。

9.一种电子设备，其特征在于，所述电子设备包括：

一个或多个处理器；

存储装置，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如权利要求1-7中任一所述的目标检测方法。

10.一种包含计算机可执行指令的存储介质，所述计算机可执行指令在由计算机处理器执行时用于执行如权利要求1-7中任一所述的目标检测方法。