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

权利要求书

1.一种图像分割方法，其特征在于，包括下述步骤：

将待分割图像划分为至少一个第一图像块；

确定每个所述第一图像块的Token，以得到Token集合；

基于所述Token集合确定每个第一图像块对应的主题；

基于所述每个第一图像块对应的主题对所述待分割图像进行分割，得到初始分割图像。

2.根据权利要求1所述的图像分割方法，其特征在于，确定每个所述第一图像块的Token，以得到Token集合包括：

基于图像特征提取模型获取每个所述第一图像块的特征向量；

对所述第一图像块的特征向量进行聚类，得到第一图像块的Token；

基于每个所述第一图像块的Token得到Token集合。

3.根据权利要求1所述的图像分割方法，其特征在于，基于所述Token集合确定每个第一图像块对应的主题包括：

构建LDA模型，并对所述LDA模型进行训练；

将所述Token集合输入训练好的所述LDA模型，得到每个第一图像块对应的主题。

4.根据权利要求1所述的图像分割方法，其特征在于，基于所述每个第一图像块对应的主题对所述待分割图像进行分割，得到初始分割图像包括：

S1、确定所述第一图像块的中心像素点，并将所述第一图像块对应的主题作为所述第一图像块的中心像素点对应的主题；

S2、对所述待分割图像进行第二次划分，得到至少一个第二图像块，以及基于所述第二图像块确定所述第二图像块的中心像素点的主题，所述第二图像块的尺寸与第一图像块的尺寸相同；

S3、继续对所述待分割图像进行划分，直至完成对所述待分割图像的W×W次划分，得到至少一个第N图像块，以及基于所述第N图像块确定所述第N图像块的中心像素点的主题，其中W为第一图像块的尺寸；

S4、基于至少一个所述第N图像块以及所述第N图像块对应的主题得到初始分割图像。

5.根据权利要求1所述的图像分割方法，其特征在于，所述图像分割方法还包括：在基于每个图像块对应的主题对所述待分割图像进行分割时，利用马尔科夫随机场模型对所述待分割图像进行约束。

6.根据权利要求4所述的图像分割方法，其特征在于，所述图像分割方法还包括：

确定所述初始分割图像的边框；

将所述初始分割图像的边框放大至预设尺寸；

对所述初始分割图像进行二进制腐蚀，得到腐蚀后的初始分割图像；

将包含腐蚀后的所述初始分割图像和边框的区域输入GraphCut模型，得到最终分割图像。

7.根据权利要求4所述的图像分割方法，其特征在于，所述图像分割方法还包括：基于所述第N图像块对应的主题对至少一个所述第N图像块进行分组，得到由所述第N图像块组成的至少一个分割对象的主题。

8.一种图像分割装置，其特征在于，包括：

划分模块，被配置为将待分割图像划分为至少一个第一图像块；

集合确定模块，被配置为确定每个所述第一图像块的Token，以得到Token集合；

主题确定模块，被配置为基于所述Token集合确定每个第一图像块对应的主题；

分割模块，被配置为基于所述每个第一图像块对应的主题对所述待分割图像进行分割，得到初始分割图像。

9.一种电子设备，包括处理器和存储装置，所述存储装置适于存储多条程序代码，其特征在于，所述程序代码适于由所述处理器加载并运行以执行权利要求1至7中任一项所述的方法。

10.一种计算机可读存储介质，其中存储有多条程序代码，其特征在于，所述程序代码适于由处理器加载并运行以执行权利要求1至7中任一项所述的方法。