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

说明书

本发明实施例公开了一种目标检测方法、装置、存储介质及电子设备，其中方法包括：获取待检测图像；将待检测图像输入到预先训练的目标检测模型中，得到目标检测模型输出的目标检测图像，其中，目标检测模型包括特征提取模块、空间注意模块和通道注意模块，所述提取模块中用于提取不同层次的特征信息，空间注意模块用于对所述特征提取模块提取的第一特征信息进行空间特征加权，通道注意模块用于对所述特征提取模块提取的第二特征信息进行通道加权，经过空间特征加权后的第一特征信息和通道加权后的第二特征信息通过图像融合得到目标检测图像。通过注意力机制关注空间特征和通道特征中的重要特征，并抑制不必要特征，有利于提高目标检测的准确度。

技术领域

本发明实施例涉及图像处理技术领域，尤其涉及一种目标检测方法、装置、存储介质及电子设备。

背景技术

显著性目标检测作为其他视觉任务的预处理步骤一直是计算机视觉领域的研究热点。可广泛应用于图像自动裁剪、图像识别和检测、视频摘要、目标跟踪等任务中。

传统的基于手工特征的显著性检测方法主要依赖手工制作的特征，而这些特征可能无法描述复杂的图像场景和结构，无法适应新的场景和对象，泛化能力差。

发明内容

本发明实施例提供一种目标检测方法、装置、存储介质及电子设备，以实现提高目标检测的检测精度。

第一方面，本发明实施例提供了一种目标检测方法，包括：

获取待检测图像；

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

第二方面，本发明实施例还提供了一种目标检测装置，包括：

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

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

第三方面，本公开实施例还提供了一种电子设备，所述电子设备包括：

一个或多个处理器；

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

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如本公开实施例任一所述的目标检测方法。

第四方面，本公开实施例还提供了一种包含计算机可执行指令的存储介质，所述计算机可执行指令在由计算机处理器执行时用于执行如本公开实施例任一所述的目标检测方法。

本实施例的技术方案，通过预先设置包括空间注意模块和通道注意模块的目标检测模型，通过目标检测模型对待检测图像进行目标检测，使得在对待检测图像进行目标检测的过程中，通过注意力机制关注空间特征和通道特征中的重要特征，并抑制不必要特征，有利于提高目标检测的准确度。

附图说明

图1为本发明实施例提供的一种目标检测方法的流程示意图；

图2是本发明实施例提供的一种目标检测模型的结构示意图；