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

说明书

本公开实施例公开了一种图像分割方法、装置、设备及存储介质。将待分割图像输入设定向量场预测模型中，获得待分割图像中各像素点的向量场；根据向量场确定各像素点所在的像素点链以及像素点链的根节点；采用预设尺寸的框对根节点进行聚合，获得聚合后的像素点链；其中，一个像素点链中的像素点组成一个第一子区域；根据第一子区域的分割边界对第一子区域进行聚合，获得多个第二子区域，得到分割后的图像。本公开实施例提供的图像分割方法，基于像素点的向量场实现对图像的分割，使得生成的超像素块携带有语义信息，且可以提高的图像分割速度。

技术领域

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

背景技术

图像分割技术是把图像分割成若干个图像块，可以应用于图像编辑中，对用户上传的图片进行自动分割，方便后续的编辑。

现有的图像分割技术，例如简单的线性迭代聚类(simple lineariterativeclustering，SLIC)算法，利用图像颜色底层特征对图像进行分割，生成的超像素不带有语义信息，且需要对参数进行调整，效率低下。其他采用深度学习的图像分割技术，耗时较大。

发明内容

本公开实施例提供一种图像分割方法、装置、设备及存储介质，以实现对图像的分割，使得生成的超像素块携带有语义信息，且可以提高的图像分割速度。

第一方面，本公开实施例提供了一种图像分割方法，包括：

将待分割图像输入设定向量场预测模型中，获得所述待分割图像中各像素点的向量场；

根据所述向量场确定各像素点所在的像素点链以及像素点链的根节点；其中，所述像素点链由满足预设条件的多个像素点组成，所述根节点为所述像素点链中的一个像素点；

采用预设尺寸的框对所述根节点进行聚合，获得聚合后的像素点链；其中，一个所述像素点链中的像素点组成一个第一子区域；

根据所述第一子区域的分割边界对所述第一子区域进行聚合，获得多个第二子区域，得到分割后的图像。

第二方面，本公开实施例还提供了一种图像分割装置，包括：

向量场获取模块，用于将待分割图像输入设定向量场预测模型中，获得所述待分割图像中各像素点的向量场；

根节点确定模块，用于根据所述向量场确定各像素点所在的像素点链以及像素点链的根节点；其中，所述像素点链由满足预设条件的多个像素点组成，所述根节点为所述像素点链中的一个像素点；

根节点聚合模块，用于采用预设尺寸的框对所述根节点进行聚合，获得聚合后的像素点链；其中，一个所述像素点链中的像素点组成一个第一子区域；

分割图像获取模块，用于根据所述第一子区域的分割边界对所述第一子区域进行聚合，获得多个第二子区域，得到分割后的图像。

第三方面，本公开实施例还提供了一种电子设备，所述电子设备包括：

一个或多个处理装置；

存储装置，用于存储一个或多个程序；

当所述一个或多个程序被所述一个或多个处理装置执行，使得所述一个或多个处理装置实现如本公开实施例所述的图像分割方法。

第四方面，本公开实施例还提供了一种计算机可读介质，其上存储有计算机程序，该程序被处理装置执行时实现如本公开实施例所述的图像分割方法。