[发明专利]物体检测方法、装置、设备及介质

说明书

本发明实施例提供了一种物体检测方法、装置、设备及介质，属于图像处理技术领域，旨在提高物体检测的准确性，所述方法包括：对待检测图像中的物体的局部区域进行检测，得到所述物体的至少一个候选局部框，其中，一个物体的候选局部框用于框选该物体的局部区域；从所述物体的至少一个候选局部框中，筛选出所述物体的目标局部框；根据所述物体的目标局部框，确定所述物体的完整框，其中，一个物体的完整框表征该物体的检测结果。

技术领域

本发明涉及图像处理技术领域，特别是涉及一种物体检测方法、装置、设备及介质。

背景技术

物体检测是利用计算机视觉技术确定图片和视频中的物体位置的技术。其中，在进行物体检测时，一般是先确定出物体的多个候选框，之后，对多个候选框去重后保留一个候选框作为物体的目标框。该目标框即框选出了物体在图像中的位置，后续便可以对目标框中的物体进行识别和处理。

然而，在一张图像中往往存在多个需要待检测的物体，此种情况下，便需要框选出多个物体的目标框。此种情况下，往往在对多个物体得到的多个候选框进行去重时，会使得一些相互有重叠的物体的目标框被漏检。例如，以物体检测中的行人检测为例，对于密集的行人检测，行人A和行人B具有遮挡关系，此时对行人A和行人B进行检测时，可能将行人B的候选框全部去除，而只保留了行人A的候选框，使得最终确定出的行人的数量不准确。

综上，现有的物体检测会导致物体检测的准确率不高的问题。

发明内容

鉴于上述问题，提出了本发明实施例的一种物体检测方法、装置、设备及介质，以便克服上述问题或者至少部分地解决上述问题。

为了解决上述问题，本发明的第一方面，公开了一种物体检测方法，所述方法包括：

对待检测图像中的物体的局部区域进行检测，得到所述物体的至少一个候选局部框，其中，一个物体的候选局部框用于框选该物体的局部区域；

从所述物体的至少一个候选局部框中，筛选出所述物体的目标局部框；

根据所述物体的目标局部框，确定所述物体的完整框，其中，一个物体的完整框表征该物体的检测结果。

可选地，从所述物体的至少一个候选局部框中筛选出一个候选局部框，作为所述物体的目标局部框，包括：

根据所述物体的至少一个候选局部框之间的重叠度，从所述物体的至少一个候选局部框中筛选出一个候选局部框，作为所述物体的目标局部框；或

根据所述物体的至少一个候选局部框分别与预先标记的所述物体的中心之间的距离，从所述物体的至少一个候选局部框中筛选出一个候选局部框，作为所述物体的目标局部框。

可选地，根据所述物体的目标局部框，确定所述物体的完整框，包括：

根据所述物体的目标局部框和所述目标局部框对应的局部特征图，确定所述物体的目标局部结构；

根据所述物体的完整结构与所述物体的各个局部结构之间的相对位置关系和所述物体的目标局部结构，确定所述物体的完整框。

可选地，根据所述物体的目标局部框，确定所述物体的完整框，包括：

将所述物体的目标局部框和所述目标局部框所框选的图像区域的特征图，输入预先训练的第一物体检测模型，得到所述物体的完整框，其中，所述第一物体检测模型是利用预先标记有样本物体的局部框和完整框的样本图像训练得到的；或

将所述物体的目标局部框和所述待检测图像的特征图，输入预先训练的第二物体检测模型，得到所述物体的完整框，其中，所述第二物体检测模型是利用预先标记有样本物体的局部框和完整框的样本图像训练得到的。

可选地，对待检测图像中的物体的局部区域进行检测，得到所述物体的至少一个候选局部框，包括：