[发明专利]一种物体识别方法、装置、终端设备及存储介质

说明书

本申请适用于人工智能技术领域，提供了一种物体识别方法、装置、终端设备及存储介质，其中，方法包括：对目标图像进行物体识别，得到至少一个候选框、每个候选框中物体的类别以及置信度；获取物体的类别的置信度大于第一预设阈值的候选框，得到第一目标候选框；对物体的类别为同一类别的第一目标候选框进行过滤，得到第二目标候选框；对所有第二目标候选框两两计算重合度，并获取重合度大于第二预设阈值的两个第二目标候选框中一个，得到第三目标候选框；获取中心位置与目标图像的预设中心位置之间的距离最短的第三目标候选框中物体的类别，得到目标图像中物体的识别结果并输出。本申请实施例可提高对目标物体识别的准确性和稳定性。

技术领域

本申请属于人工智能技术领域，尤其涉及一种物体识别方法、装置、终端设备及存储介质。

背景技术

随着人工智能技术的快速发展，各种人工智能产品顺应而生，人工智能产品可对特定的物体进行物体识别，以识别出物体的类别和位置信息。

目前基于深度学习的物体识别大多数采用分类算法，容易把背景误识别成物体以及识别出画面中的多个物体，不能准确出的确定要识别目标物体，使得对目标物体识别的准确性和稳定性不高。

发明内容

本申请实施例提供了一种物体识别方法、装置、终端设备及存储介质，旨在解决现有对目标物体识别的准确性和稳定性不高的问题。

第一方面，本申请实施例提供了一种物体识别方法，包括：

获取目标图像，检测所述目标图像的图像质量；

当所述目标图像的图像质量满足第一预设条件时，对所述目标图像进行物体识别，得到至少一个候选框、每个所述候选框中物体的类别以及每个所述候选框中物体的类别的置信度；

从所有所述候选框中获取物体的类别的置信度大于第一预设阈值的候选框，得到第一目标候选框；

对物体的类别为同一类别的第一目标候选框进行过滤，得到第二目标候选框；其中，得到的每两个第二目标候选框中的物体的类别不同；

对所有所述第二目标候选框两两计算重合度，并获取重合度大于第二预设阈值的两个所述第二目标候选框中一个，得到第三目标候选框；

当得到多个所述第三目标候选框时，获取中心位置与所述目标图像的预设中心位置之间的距离最短的第三目标候选框中物体的类别，得到所述目标图像中物体的识别结果并输出。

第二方面，本申请实施例提供了一种物体识别装置，包括：

检测模块，用于获取目标图像，检测所述目标图像的图像质量；

识别模块，用于当所述目标图像的图像质量满足第一预设条件时，对所述目标图像进行物体识别，得到至少一个候选框、每个所述候选框中物体的类别以及每个所述候选框中物体的类别的置信度；

第一获取模块，用于从所有所述候选框中获取物体的类别的置信度大于第一预设阈值的候选框，得到第一目标候选框；

过滤模块，用于对物体的类别为同一类别的第一目标候选框进行过滤，得到第二目标候选框；其中，得到的每两个第二目标候选框之间的物体的类别不同；

第二获取模块，用于对所有所述第二目标候选框两两计算重合度，并获取重合度大于第二预设阈值的两个所述第二目标候选框中一个，得到第三目标候选框；

第三获取模块，用于当得到多个所述第三目标候选框时，获取中心位置与所述目标图像的预设中心位置之间的距离最短的第三目标候选框中物体的类别，得到所述目标图像中物体的识别结果并输出。

第三方面，本申请实施例提供了一种终端设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现上述物体识别方法的步骤。