[发明专利]一种基于局部关系指导的图像分割方法

说明书

本发明公开了一种基于局部关系指导的图像分割方法，包括如下步骤：S1、在卷积神经网络模型中设计两个分支，第一分支为异同判断模块，第二分支保留原有的分割模块；S2、异同判断模块输出得到待分割图像的局部关系图r，并通过已有分割标注信息的变换进行监督；S3、分割模块输出初步分割结果S4、将局部关系图r与初步分割结果融合，通过定义的引导公式进行引导，调整初步分割结果生成最终分割结果p。本发明将像素局部关系信息作用到初步分割结果中，使得像素点的输出能显式利用周围像素点的分类信息和周围像素点与中心像素点的关系信息，克服了现有技术无法有效利用边界信息的缺陷，提高图像分割的准确性，同时有效降低了学习的难度。

技术领域

本发明涉及图像分割技术领域，特别涉及一种基于局部关系指导的图像分割方法。

背景技术

图像分割是由图像处理到图像分析的关键步骤，将场景中的各元素各物体的边界准确分割出来，一直是图像分割任务的一大挑战。由于场景分割结果中包含了各个元素的形状信息，这些信息反映了图像的结构信息，因此结构信息在场景分割中相当重要。现有的利用结构信息进行图像分割的策略主要有：1、直接利用注意力机制来保留结构信息；2、通过计算某一个像素点与全图每个像素点的相似度得到相似图，再利用注意力机制找到感兴趣区域。然而，有工作表明，注意力机制激活的区域近乎全图，并没有关注同类别区域，进而无法实现结构保留的目的，并且空间注意力机制还存在计算量大的缺点。

发明内容

为解决上述问题，本发明提供了一种基于局部关系指导的图像分割方法。

本发明采用以下技术方案：

一种基于局部关系指导的图像分割方法，包括如下步骤：

S1、在卷积神经网络模型中设计两个分支，第一分支为异同判断模块，用于产生局部关系图，第二分支保留原有的分割模块，用于获得初步分割结果，所述第一分支和所述第二分支共享所述卷积神经网络的骨干网络；

S2、所述异同判断模块输出得到待分割图像的局部关系图r，所述局部关系图r∈R^H*W*k*k，其中，r是一个H*W*k*k的矩阵，H、W分别表示待分割图像的高与宽，该矩阵表示每个像素点与以自己为中心、k*k窗口内其他像素点的异同类别关系，所述异同判断模块的输出结果通过已有分割标注信息的变换进行监督；

S3、所述分割模块输出初步分割结果所述初步分割结果其中，C表示待分割图像中的分割对象的类别数，H、W分别表示待分割图像的高与宽；

S4、将所述局部关系图r与所述初步分割结果融合，通过引导公式进行引导，调整所述初步分割结果生成最终分割结果p。

进一步地，所述步骤S2中的所述异同类别关系用概率表示，表示像素i与像素j的异同类别的概率，反映了i与j的异同关系。

进一步地，所述步骤S3中的分割结果得到的每个像素点的分类概率值记为表示像素点i分类为第c类的概率值。

进一步地，所述步骤S4中的所述引导公式为：

其中，p_i,j为经引导后的概率值，为初步分割结果的每个像素点的分类概率值，γ∈[0,1)，是人为设定的超参数，即认定异同性概率输出大于γ即认为是同类别，为指示函数，即当x为真时，否则为0；ω为以i，j为中心的一个小邻域(如3*3)，δ为遍历该邻域的变量；

该引导公式以像素i为中心，将像素i周围ω范围内的与之相同类别的像素点的概率值汇聚到中心，与之不同类别的像素点，则不予考虑。

采用上述技术方案后，本发明与背景技术相比，具有如下优点：