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

说明书

本公开提供了一种目标检测方法、装置、电子设备及存储介质，其中，该目标检测方法包括：获取待检测图像；基于所述待检测图像，确定角点在所述待检测图像中的角点位置信息以及各个角点对应的向心偏移张量，所述角点表征所述待检测图像中的目标对象的位置；基于各个角点在所述待检测图像中的角点位置信息及各个角点对应的向心偏移张量，确定所述待检测图像中的目标对象。

技术领域

本公开涉及图像识别技术领域，具体而言，涉及一种目标检测方法、装置、电子设备及存储介质。

背景技术

目标检测是计算机视觉重要的基础问题，许多计算机视觉的应用都依赖于目标检测，比如自动驾驶、视频监控和移动娱乐。

在进行目标检测时，主要任务是用检测框标出图像中物体所在位置，该过程可以基于物体关键点的目标检测算法确定物体在图像中的位置，通过确定出图像中所有物体关键点后，将属于相同物体的物体关键点进行匹配，从而得到物体的检测框。

当图像中包含多个外观相似的物体时，因外观相似的物体对应的物体关键点之间的匹配度较高，容易造成错误的检测结果，比如检测结果为同一个检测框中包含了多个物体，因此，目前的目标检测方法的检测精度准确度较低。

发明内容

本公开实施例至少提供一种目标检测方案。

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

获取待检测图像；

基于所述待检测图像，确定角点在所述待检测图像中的角点位置信息以及各个角点对应的向心偏移张量，所述角点表征所述待检测图像中的目标对象的位置；

基于各个角点在所述待检测图像中的角点位置信息及各个角点对应的心偏移张量，确定所述待检测图像中的目标对象。

本公开提供了一种目标检测方法，在获取到待检测图像后，首先确定角点在待检测图像中的角点位置信息以及各个角点对应的向心偏移张量，因为角点是指图像中的主要特征点，其在待检测图像中的角点位置信息能够表征每个目标对象在待检测图像中的位置，比如角点可以包括左上角点和右下角点，其中左上角点是指对应目标对象上侧轮廓的直线与对应目标对象左侧轮廓的直线的交点，右下角点是指对应目标对象下侧轮廓的直线与对应目标对象右侧轮廓的直线的交点，当左上角点和右下角点属于同一个目标对象的检测框时，左上角点和右下角点分别对应的向心偏移张量指向的位置应该比较接近，因此，本公开实施例提出的目标检测方法，基于各个角点在待检测图像中的角点位置信息及各个角点对应的向心偏移张量，能够确定出属于同一目标对象的角点，进而基于确定出的角点可以检测出该同一目标对象。

在一种可能的实施方式中，所述基于所述待检测图像，确定角点在待检测图像中的角点位置信息以及各个角点对应的向心偏移张量，包括：

对所述待检测图像进行特征提取，得到所述待检测图像对应的初始特征图；

对所述初始特征图进行角点池化处理，得到角点池化后的特征图；

基于所述角点池化后的特征图，确定各个角点在所述待检测图像中的角点位置信息，以及各个角点对应的向心偏移张量。

这里通过对待检测图像进行特征提取，得到初始特征图，并对初始特征图进行角点池化处理，得到能够便于提取角点以及角点对应的向心偏移量的特征图，即角点池化后的特征图。

在一种可能的实施方式中，所述基于所述角点池化后的特征图，确定各个角点在所述待检测图像中的角点位置信息，包括：

基于所述角点池化后的特征图，生成所述待检测图像对应的角点热力图；

基于所述角点热力图，确定所述角点热力图中每个特征点作为角点的概率值，并基于每个特征点作为角点的概率值，从所述特征点中筛选出所述角点；