[发明专利]一种物体检测方法、物体检测装置及电子设备

说明书

本申请公开了一种物体检测方法、装置、电子设备及计算机可读存储介质。其中，该方法包括：将待检测图像输入已训练的目标检测模型中，得到已训练的目标检测模型所返回的至少一个预估包含目标物体的检测框；确定各个检测框所分别对应的区域图像；针对每个区域图像，在计算所得到的颜色面积中，确定面积最大值，其中，颜色面积包括至少一个预设颜色分别在区域图像中所占的面积，预设颜色为目标物体能够存在的颜色；将面积最大值大于预设的面积阈值的区域图像确定为候选区域图像；将距离待检测图像的预设位置点最近的候选区域图像确定为真实显示有目标物体的目标区域图像。通过本方案，可以提高对形状相同但颜色不同的物体的识别准确率。

技术领域

本申请属于图像处理技术领域，尤其涉及一种物体检测方法、物体检测装置、电子设备及计算机可读存储介质。

背景技术

通常来说，流水线作业的任务比较单一，一条流水线往往专注于生产同一型号的产品。当前，大部分厂商为了给用户提供更多样化及个性化的产品选择，会在一个产品型号下开发出不同颜色，这使得流水线上所生产的同一产品型号的不同产品可能有着不同的颜色，给对流水线上的产品的识别带来了一定难度。

发明内容

本申请提供了一种物体检测方法、物体检测装置、电子设备及计算机可读存储介质，可以解决对流水线上有着多种颜色选择的物体(也即流水线上形状相同但颜色不同的物体)的识别较为困难的问题。

第一方面，本申请提供了一种物体检测方法，包括：

将待检测图像输入已训练的目标检测模型中，得到上述已训练的目标检测模型所返回的至少一个预估包含目标物体的检测框；

确定各个检测框所分别对应的区域图像，其中，任一区域图像的外边界为对应的检测框；

针对每个区域图像，在计算所得到的颜色面积中，确定面积最大值，其中，上述颜色面积包括至少一个预设颜色分别在上述区域图像中所占的面积，上述预设颜色为上述目标物体能够存在的颜色；

将上述面积最大值大于预设的面积阈值的区域图像确定为候选区域图像；

将距离上述待检测图像的预设位置点最近的候选区域图像确定为真实显示有上述目标物体的目标区域图像。

第二方面，本申请提供了一种物体检测装置，包括：

目标检测单元，用于将待检测图像输入已训练的目标检测模型中，得到上述已训练的目标检测模型所返回的至少一个预估包含目标物体的检测框；

区域图像确定单元，用于确定各个检测框所分别对应的区域图像，其中，任一区域图像的外边界为对应的检测框；

面积最大值确定单元，用于针对每个区域图像，在计算所得到的颜色面积中，确定面积最大值，其中，上述颜色面积包括至少一个预设颜色分别在上述区域图像中所占的面积，上述预设颜色为上述目标物体能够存在的颜色；

候选区域图像确定单元，用于将上述面积最大值大于预设的面积阈值的区域图像确定为候选区域图像；

目标区域图像确定单元，用于将距离上述待检测图像的预设位置点最近的候选区域图像确定为真实显示有上述目标物体的目标区域图像。

第三方面，本申请提供了一种电子设备，上述电子设备包括存储器、处理器以及存储在上述存储器中并可在上述处理器上运行的计算机程序，上述处理器执行上述计算机程序时实现如上述第一方面的方法的步骤。

第四方面，本申请提供了一种计算机可读存储介质，上述计算机可读存储介质存储有计算机程序，上述计算机程序被处理器执行时实现如上述第一方面的方法的步骤。

第五方面，本申请提供了一种计算机程序产品，上述计算机程序产品包括计算机程序，上述计算机程序被一个或多个处理器执行时实现如上述第一方面的方法的步骤。