[发明专利]一种图像分割方法和电子设备

权利要求书

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

图像分割模型的编码器根据对应的编码器参数对目标图像进行特征提取处理，得到所述目标图像的图像特征；所述图像分割模型包括所述编码器和主解码器；

所述主解码器根据对应的解码器参数和所述图像特征，进行图像边界分割处理，得到所述目标图像的边界分割信息；

其中，所述编码器对应的编码器参数，根据与预定图像的图像特征匹配的图像边界分割信息确定；所述预定图像的图像特征包括：已标注的第一样本图像的第一图像特征，未标注的第二样本图像的第二图像特征和对所述第二图像特征进行扰动处理所得的第三图像特征。

2.根据权利要求1所述的方法，所述图像分割模型的编码器根据对应的编码器参数对目标图像进行特征提取处理，得到目标图像的图像特征，包括：

所述编码器根据对应的编码器参数提取所述目标图像的特征，得到所述目标图像的第一尺寸的特征图；所述第一尺寸小于所述目标图像的第二尺寸；

所述边界分割信息包括所述第二尺寸的特征图中各像元对应的分类类别信息；对像元的分类处理用于将像元分类为是否属于图像中对象的边界；

所述主解码器根据对应的解码器参数和所述图像特征，进行图像边界分割处理，得到所述目标图像的边界分割信息，包括：

所述主解码器根据对应的解码器参数将所述第一尺寸的特征图映射到所述第二尺寸，得到第二尺寸的特征图；

所述主解码器对所述第二尺寸的特征图中的像元进行分类处理，得到所述第二尺寸的特征图中各像元的分类类别信息。

3.根据权利要求2所述的方法，所述编码器对所述目标图像执行有多次卷积和下采样操作；其中，不同的下采样操作得到不同尺寸的特征图，最后一次下采样操作得到所述第一尺寸的特征图；

所述主解码器对所述第一尺寸的特征图执行有多次卷积和上采样操作；其中，不同的上采样操作得到不同尺寸的特征图，最后一次上采样操作得到所述第二尺寸的特征图；

所述主解码器针对所述第一尺寸的特征图在非第一次执行卷积操作前，将最近上采样得到的当前特征图与编码器得到的与所述当前特征图尺寸相同的特征图进行连接处理，并对连接处理所得的连接特征图进行卷积操作。

4.根据权利要求1所述的方法，所述图像分割模型的构建过程包括：

利用与所述第一样本图像的第一图像特征匹配的图像边界分割信息优化预定网络模型的第一分支；所述第一分支包括编码器模型和与所述编码器模型的输出端相连的主解码器模型；

利用与所述第二样本图像的第二图像特征匹配的图像边界分割信息，和与所述第二样本图像的第三图像特征匹配的图像边界分割信息，优化所述预定网络模型的第二分支；所述第二分支包括所述编码器模型和与所述编码器模型的输出端相连的辅助解码器模型；完成优化的所述编码器模型和所述主解码器模型，分别作为所述图像分割模型的编码器和主解码器。

5.根据权利要求4所述的方法，所述利用与所述第一样本图像的第一图像特征匹配的图像边界分割信息优化预定网络模型的第一分支，包括：

获得基于所述第一分支对所述第一样本图像进行特征提取，并对提取的第一图像特征进行处理得到的第一边界分割信息；

获得为所述第一样本图像标注的第二边界分割信息；

根据所述第一边界分割信息和所述第二边界分割信息，调整所述编码器模型对应的编码器参数和所述主解码器模型对应的解码器参数。

6.根据权利要求4所述的方法，所述利用与所述第二样本图像的第二图像特征匹配的图像边界分割信息，和与所述第二样本图像的第三图像特征匹配的图像边界分割信息，优化所述预定网络模型的第二分支，包括：

获得基于所述第一分支对所述第二样本图像进行特征提取，并对提取的第二图像特征进行处理得到的第三边界分割信息；

获得基于所述第二分支对所述第二样本图像进行特征提取，对提取的第二图像特征进行扰动处理，并对扰动后得到的第三图像特征进行处理得到的第四边界分割信息；

根据所述第三边界分割信息和所述第四边界分割信息，调整所述编码器模型对应的编码器参数和所述辅助解码器模型对应的解码器参数。