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

说明书

本申请公开了目标检测方法、装置、电子设备及存储介质，具体实现方案为：对待检测图像进行特征提取，得到多尺度特征图；将所述多尺度特征图输入训练好的目标检测网络，得到第一子特征图及第二子特征图；其中，所述第一子特征图用于表征所述待检测图像的中心区域特征，所述第二子特征图用于表征所述待检测图像的边框区域特征；在所述训练好的目标检测网络中，将所述第一子特征图及所述第二子特征图采用基于中心区域预测的回归方式进行预测，输出多个候选检测框；根据所述多个候选检测框，得到目标检测框。采用本申请，可以提高目标检测的准确率。

技术领域

本申请涉及计算机视觉技术领域，尤其涉及一种目标检测方法、装置、电子设备及存储介质。

背景技术

随着便携设备、手机终端等电子设备相比以往更智能化，芯片的解析能力更强，可以通过计算机视觉技术对图文信息、视频信息等进行高效的解析，并对图文信息、视频信息等中的目标对象进行检测。

目标检测算法大多是采用矩形方框作为目标检测框，并检测图像中规则形状的图文信息，然而，在实际应用中存在倾斜角度或不规则的图像，或者图像中存在不规则形状的图文信息等，在这些情况下，如何提高目标检测准确率，未存在有效的解决方案。

发明内容

本申请提供了一种目标检测方法、装置、电子设备及存储介质。

根据本申请的一方面，提供了一种目标检测方法，包括：

对待检测图像进行特征提取，得到多尺度特征图；

将所述多尺度特征图输入训练好的目标检测网络，得到第一子特征图及第二子特征图；其中，所述第一子特征图用于表征所述待检测图像的中心区域特征，所述第二子特征图用于表征所述待检测图像的边框区域特征；

在所述训练好的目标检测网络中，将所述第一子特征图及所述第二子特征图采用基于中心区域预测的回归方式进行预测，输出多个候选检测框；

根据所述多个候选检测框，得到目标检测框。

根据本申请的另一方面，提供了一种目标检测装置，包括：

特征提取模块，用于对待检测图像进行特征提取，得到多尺度特征图；

特征分类模块，用于将所述多尺度特征图输入训练好的目标检测网络，得到第一子特征图及第二子特征图；其中，所述第一子特征图用于表征所述待检测图像的中心区域特征，所述第二子特征图用于表征所述待检测图像的边框区域特征；

目标检测模块，用于在所述训练好的目标检测网络中，将所述第一子特征图及所述第二子特征图采用基于中心区域预测的回归方式进行预测，输出多个候选检测框；

根据所述多个候选检测框，得到目标检测框。

根据本申请的另一方面，提供了一种电子设备，包括：

至少一个处理器；以及

与该至少一个处理器通信连接的存储器；其中，

该存储器存储有可被该至少一个处理器执行的指令，该指令被该至少一个处理器执行，以使该至少一个处理器能够执行本申请任意一实施例所提供的方法。

根据本申请的另一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，该计算机指令用于使该计算机执行本申请任意一项实施例所提供的方法。