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

说明书

本公开提出了一种目标物体的检测方法、装置、电子设备和存储介质，涉及人工智能技术领域，尤其涉及计算机视觉和深度学习技术领域，可应用于智能交通场景。具体实现方案为：获取待识别的图像，根据待识别的图像，生成所述待识别图像的多个尺度特征图，将多个尺度特征图，输入目标框检测模型的头部网络，以得到待识别图像中目标物体的水平框的位置信息和目标物体的目标框的方位信息，根据水平框的位置信息和目标物体的目标框的方位信息，得到待识别图像中目标物体的目标框位置信息，目标框位置信息包含目标框的位置和目标框的方向信息，实现在定位旋转框的位置的同时，准确定位旋转框的方向信息，提高了目标物体定位的准确性。

技术领域

本公开涉及人工智能技术领域，尤其涉及计算机视觉和深度学习技术领域，可应用于智能交通场景，具体涉及目标物体的检测方法、装置、电子设备和存储介质。

背景技术

通用目标检测方法用垂直的边框定位图像中的目标(例如人和车等)。而实际应用中，图像中的目标物体，并不一定是垂直的，可能存在偏斜，因此，垂直的边框无法准确定位图像中方向多变的目标物体，例如，智能交通场景下的目标物体，如车辆、飞机等，因此，如何准确定位图像中的目标物体是亟待解决的技术问题。

发明内容

本公开提供了一种用于提高目标物体检测效果的目标物体的检测方法、装置、电子设备和存储介质。

根据本公开的一方面，提供了一种目标物体的检测方法，其中，包括：

获取待识别的图像；

根据所述待识别的图像，生成所述待识别图像的多个尺度特征图；

将所述多个尺度特征图，输入目标框检测模型的头部网络，以得到所述待识别图像中目标物体的水平框的位置信息和目标物体的目标框的方位信息；

根据所述水平框的位置信息和所述目标框的方位信息，得到所述待识别图像中目标物体的目标框位置信息；所述目标框位置信息包含所述目标框的位置和所述目标框的方向信息。

根据本公开的另一方面，提供了一种目标物体的检测装置，其中，包括：

获取模块，用于获取待识别的图像；

生成模块，用于根据所述待识别的图像，生成所述待识别图像的多个尺度特征图；

处理模块，用于将所述多个尺度特征图，输入目标框检测模型的头部网络，以得到所述待识别图像中目标物体的水平框的位置信息和目标物体的目标框的方位信息；

确定模块，用于根据所述水平框的位置信息和所述目标框的方位信息，得到所述待识别图像中目标物体的目标框位置信息；所述目标框位置信息包含所述目标框的位置和所述目标框的方向信息。

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

至少一个处理器；以及

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

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行第一方面所述的方法。

根据本公开的另一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其中，所述计算机指令用于使所述计算机执行第一方面所述的方法。

根据本公开的另一方面，提供了一种计算机程序产品，包括计算机程序，所述计算机程序在被处理器执行时实现第一方面所述的方法。

应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征，也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。

附图说明

附图用于更好地理解本方案，不构成对本公开的限定。其中：