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

权利要求书

1.一种目标物体的检测方法，其中，包括：

获取待识别的图像；

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

将每一个尺度特征图，输入对应的头部网络，以得到所述头部网络输出的目标物体的候选水平框的位置信息和目标物体的候选目标框的方位信息；

获取所述多个头部网络的权重；

根据每一个头部网络的权重，和所述每一个头部网络输出的目标物体的候选水平框的位置信息和目标物体的候选目标框的方位信息，确定所述待识别图像中目标物体的水平框的位置信息和目标物体的目标框的方位信息；所述头部网络为多个，每一个头部网络对应一个尺度特征图；

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

2.如权利要求1所述的方法，其中，将所述多个尺度特征图，输入目标框检测模型的头部网络，以得到所述待识别图像中目标物体的水平框的位置信息，包括：

目标框检测模型的头部网络，根据所述多个尺度特征图，得到所述多个尺度特征图中每个位置点的偏移量信息；

目标框检测模型的头部网络，根据所述每个位置点的偏移量信息，确定所述待识别图像中目标物体的水平框的位置信息。

3.如权利要求1-2任一所述的方法，其中，所述方位信息包含所述目标物体的目标框与所述水平框的位置偏移量，以及所述目标物体的目标框的朝向信息。

4.如权利要求1所述的方法，其中，所述方法还包括：

将多个尺度特征图，输入和每一个尺度特征图对应的目标框检测模型的头部网络，以确定所述目标物体的类别。

5.一种目标物体的检测装置，其中，包括：

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

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

处理模块，用于将每一个尺度特征图，输入对应的头部网络，以得到所述头部网络输出的目标物体的候选水平框的位置信息和目标物体的候选目标框的方位信息；获取所述多个头部网络的权重；根据每一个头部网络的权重，和所述每一个头部网络输出的目标物体的候选水平框的位置信息和目标物体的候选目标框的方位信息，确定所述待识别图像中目标物体的水平框的位置信息和目标物体的目标框的方位信息；所述头部网络为多个，每一个头部网络对应一个尺度特征图；

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

6.如权利要求5所述的装置，其中，所述处理模块，用于：

目标框检测模型的头部网络，根据所述多个尺度特征图，得到所述多个尺度特征图中每个位置点的偏移量信息；

目标框检测模型的头部网络，根据所述每个位置点的偏移量信息，确定所述待识别图像中目标物体的水平框的位置信息。

7.如权利要求5-6任一所述的装置，其中，所述方位信息包含所述目标物体的目标框与所述水平框的位置偏移量，以及所述目标物体的目标框的朝向信息。

8.如权利要求5所述的装置，其中，所述处理模块，还用于：

将多个尺度特征图，输入和每一个尺度特征图对应的目标框检测模型的头部网络，以确定所述目标物体的类别。

9.一种电子设备，包括：

至少一个处理器；以及

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

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行权利要求1-4中任一项所述的方法。

10.一种存储有计算机指令的非瞬时计算机可读存储介质，其中，所述计算机指令用于使所述计算机执行根据权利要求1-4中任一项所述的方法。