[发明专利]图像分割方法及装置、电子设备和存储介质

说明书

本公开涉及一种图像分割方法及装置、电子设备和存储介质。所述方法包括：获取待处理图像的第一待处理特征；根据所述第一待处理特征，得到所述待处理图像的第一边缘预测图；根据所述第一边缘预测图，对所述第一待处理特征进行调整，得到调整后的第一待处理特征；根据所述调整后的第一待处理特征，得到所述待处理图像的第一分割预测图。

技术领域

本公开涉及图像处理技术领域，尤其涉及一种图像分割方法及装置、电子设备和存储介质。

背景技术

镜子、玻璃、透明的杯子等反射率较高的物体或透明体广泛地分布在我们的现实生活中。对图像中的这类物体进行识别有着广泛的应用，例如，在机器人的视觉导航中，机器人需要避开路径中的这类障碍物。对图像中的这类物体进行分割，是在图像中识别这类物体的至关重要的预处理步骤。图像分割的准确性将影响图像识别的准确性。

发明内容

本公开提供了一种图像分割技术方案。

根据本公开的一方面，提供了一种图像分割方法，包括：

获取待处理图像的第一待处理特征；

根据所述第一待处理特征，得到所述待处理图像的第一边缘预测图；

根据所述第一边缘预测图，对所述第一待处理特征进行调整，得到调整后的第一待处理特征；

根据所述调整后的第一待处理特征，得到所述待处理图像的第一分割预测图。

通过获取待处理图像的第一待处理特征，根据所述第一待处理特征，得到所述待处理图像的第一边缘预测图，根据所述第一边缘预测图，对所述第一待处理特征进行调整，得到调整后的第一待处理特征，并根据所述调整后的第一待处理特征，得到所述待处理图像的第一分割预测图，由此通过增强对边缘信息的处理，能够提高对所述待处理图像进行图像分割的准确性。

在一种可能的实现方式中，所述根据所述第一待处理特征，得到所述待处理图像的第一边缘预测图，包括：

获取所述待处理图像的第一浅层特征；

根据所述第一待处理特征，以及所述待处理图像的第一浅层特征，得到所述待处理图像的第一边缘预测图。

在该实现方式中，通过融合所述待处理图像的第一待处理特征和所述待处理图像的第一浅层特征，对所述待处理图像进行边缘预测，由此能够结合所述待处理图像的第一浅层特征中的更丰富的边缘信息，得到更准确的所述待处理图像的第一边缘预测图。

在一种可能的实现方式中，所述根据所述第一边缘预测图，对所述第一待处理特征进行调整，得到调整后的第一待处理特征，包括：

根据所述第一边缘预测图，确定K个第一关键点，其中，K为大于1的整数；

对所述第一待处理特征中，所述K个第一关键点的初始特征进行调整，得到调整后的第一待处理特征。

在该实现方式中，通过根据所述第一边缘预测图，确定K个第一关键点，并对所述第一待处理特征中，所述K个第一关键点的初始特征进行调整，由此得到的调整后的第一待处理特征能够更准确地表达全局的边缘信息，从而有利于进一步提高对所述待处理图像进行图像分割的准确性。

在一种可能的实现方式中，所述根据所述第一边缘预测图，确定K个第一关键点，包括：

将所述第一边缘预测图中属于边缘的概率最高的K个像素点，确定为K个第一关键点。

通过将所述第一边缘预测图中属于边缘的概率最高的K个像素点，确定为K个第一关键点，由此确定出的K个第一关键点能够更准确地表示所述待处理图像的边缘的信息。

在一种可能的实现方式中，所述对所述第一待处理特征中，所述K个第一关键点的初始特征进行调整，得到调整后的第一待处理特征，包括：