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

权利要求书

1.一种图像分割的方法，包括：

获取包含前景的待分割图像；

将所述待分割图像输入至预先训练的图像分割模型，生成所述待分割图像对应的分割结果图，其中，所述图像分割模型中包括编码器、解码器和连接层，所述连接层用于将所述编码器的隐层所提取的特征输入至所述解码器的隐层，所述图像分割模型基于语义连通性损失训练得到，所述语义连通性损失基于训练样本包括的样本图像分割结果图中包括的至少一个连通域与图像分割结果预测图中包括的目标数目个连通域的匹配而确定，所述样本图像分割结果图中包括的至少一个连通域与前景相匹配，所述图像分割结果预测图通过将所述训练样本包括的样本图像输入至所述图像分割模型而得到。

2.根据权利要求1所述的方法，其中，所述编码器中包括卷积层和第一数目个通道混洗卷积单元，所述通道混洗卷积单元用于重组通过不同组的分组卷积所提取的特征，所述解码器中包括第二数目个深度可分离卷积单元，所述连接层用于将所述卷积层输出的特征输入至所述解码器的深度可分离卷积单元；以及

所述将所述待分割图像输入至预先训练的图像分割模型，生成所述待分割图像对应的分割结果图，包括：

将所述待分割图像输入至所述卷积层，生成第一特征图；

将所述第一特征图输入至所述第一数目个通道混洗卷积单元，生成第二特征图；

将所述第一特征图输入至所述连接层，生成第三特征图；

将所述第二特征图和所述第三特征图输入至所述第二数目个深度可分离卷积单元，生成所述分割结果图。

3.根据权利要求2所述的方法，其中，所述第一数目个通道混洗卷积单元被划分为串联连接的第三数目个特征提取阶段，所述第三数目和各特征提取阶段中不同层所对应的通道的数目分别小于各自对应的预设阈值；以及

所述将所述第二特征图和所述第三特征图输入至所述第二数目个深度可分离卷积单元，生成所述分割结果图，包括：

将第二特征图输入至解码器包括的第一个深度可分离卷积单元，生成第四特征图；

对所述第四特征图进行上采样，生成与所述第三特征图的分辨率一致的第五特征图；

将所述第五特征图与所述第三特征图进行融合；

将融合后的特征图输入至所述解码器包括的更深层次的深度可分离卷积单元，生成所述分割结果图。

4.根据权利要求1-3之一所述的方法，其中，所述图像分割模型基于训练样本训练得到，所述训练样本包括的样本图像通过以下步骤得到：

获取原始尺寸的初始样本图像；

按照与所述原始尺寸的宽高比一致的纵横比对所述初始样本图像进行缩放，生成不超过预设尺寸的缩放图；

对所述预设尺寸与所述缩放图之间的空白进行填充，生成所述样本图像。

5.根据权利要求1-3之一所述的方法，其中，所述前景包括人像。

6.根据权利要求5所述的方法，其中，所述语义连通性损失通过以下步骤而确定：

对于所述样本图像分割结果图中包括的至少一个连通域中的连通域，基于该连通域与所述图像分割结果预测图中包括的目标数目个连通域的匹配，生成该连通域对应的连通值；

根据所生成的与所述至少一个连通域分别对应的连通值，生成语义连通性损失值。

7.根据权利要求6所述的方法，其中，所述基于该连通域与所述图像分割结果预测图中包括的目标数目个连通域的匹配，生成该连通域对应的连通值，包括：

对于所述样本图像分割结果图中包括的至少一个连通域中的连通域，基于该连通域与所述图像分割结果预测图中相交的连通域的交并比，生成该连通域对应的连通值；

将所生成的至少一个连通域进行求和；

将求和结果与目标值的比值确定为连通总值，其中，所述目标值基于所述样本图像分割结果图和所述图像分割结果预测图中的相交的连通域和未相交的连通域的数目而确定；

利用与所述连通总值负相关的语义连通损失函数根据所述连通总值生成所述语义连通性损失值。