[发明专利]图像识别方法、目标对象的提取方法、装置、介质及设备

说明书

本发明公开了图像识别方法、目标对象的提取方法、装置、介质及设备，其中图像识别方法包括：获取待识别图像，其中，待识别图像中包括至少一个目标对象；将待识别图像输入至预先训练的识别模型中，基于识别模型的输出结果确定待识别图像中各目标对象的位置信息以及各目标对象的层叠信息，其中，识别模型基于目标对象的样本图像和样本图像中各目标对象的层叠标签和位置标签训练得到。通过具有位置信息和层叠信息识别功能的识别模型对待识别图像进行处理，得到各目标对象的位置信息和层叠信息，通过层叠信息辅助位置信息，对各目标对象进行定位，提高定位信息的准确性，同时基于层叠信息确定目标对象的被遮挡情况，便于对目标对象进行高精度的提取。

技术领域

本发明实施例涉及图像处理技术领域，尤其涉及图像识别方法、目标对象的提取方法、装置、介质及设备。

背景技术

在诸如茶叶、花瓣等能够层叠放置的物体的处理过程中，包括对分布的上述物体进行回收提取的过程，目前的技术方案中，主要是通过对上述分布的物体进行图像定位的方式，确定各物体的位置，以对各物体进行回收提取。

但是目前的图像定位方式，存在定位精度差导致上述物体回收率低的问题。

发明内容

本发明提供一种图像识别方法、装置、存储介质及电子设备，以实现目标对象的精准定位。

第一方面，本发明实施例提供了一种图像识别方法，包括：

获取待识别图像，其中，所述待识别图像中包括至少一个目标对象；

将所述待识别图像输入至预先训练的识别模型中，基于所述识别模型的输出结果确定所述待识别图像中各目标对象的位置信息以及各所述目标对象的层叠信息，其中，所述识别模型基于目标对象的样本图像和所述样本图像中各目标对象的层叠标签和位置标签训练得到。

第二方面，本发明实施例还提供了一种目标对象的提取方法，包括：

获取至少一个目标对象的第一图像；

将所述第一图像输入至预先训练的识别模型中，基于所述识别模型的输出结果确定所述第一图像中各目标对象的位置信息以及各所述目标对象的层叠信息；

基于各所述目标对象的位置信息和层叠信息对各目标对象进行提取。

第三方面，本发明实施例还提供了一种图像识别装置，包括：

第一图像获取模块，用于获取待识别图像，其中，所述待识别图像中包括至少一个目标对象；

第一图像识别模块，用于将所述待识别图像输入至预先训练的识别模型中，基于所述识别模型的输出结果确定所述待识别图像中各目标对象的位置信息以及各所述目标对象的层叠信息，其中，所述识别模型基于目标对象的样本图像和所述样本图像中各目标对象的层叠标签和位置标签训练得到。

第四方面，本发明实施例还提供了一种目标对象的提取装置，包括：

第二图像获取模块，用于获取至少一个目标对象的第一图像；

第二图像识别模块，用于将所述第一图像输入至预先训练的识别模型中，基于所述识别模型的输出结果确定所述第一图像中各目标对象的位置信息以及各所述目标对象的层叠信息；

目标对象提取模块，用于基于各所述目标对象的位置信息和层叠信息对各目标对象进行提取。

第五方面，本发明实施例还提供了一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现如本发明任一实施例提供的图像识别方法，或目标对象的提取方法。

第六方面，本发明实施例还提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现如本发明任一实施例提供的图像识别方法，或目标对象的提取方法。