[发明专利]图像分割方法及相关装置

说明书

本申请实施例公开了一种图像分割方法及相关装置，应用于电子设备，电子设备预先配置图像分割网络模型，其包括编码模块、特征平衡模块和解码模块，方法包括：将输入图像输入到编码模块进行特征提取，得到特征图像，并对特征图像逐步进行下采样操作，得到N层图像特征，输入图像为交通场景图像；将N层图像特征输入到特征平衡模块，得到融合后的N个融合图像特征，特征平衡模块用于融合不同层次的图像特征；将N个融合图像特征输入到解码模块进行上采样，得到N个上采样图像，每一特征融合图像对应一个上采样图像；将N个上采样图像与N个融合图像特征进行拼接融合，得到图像分割结果。采用本申请实施例可以提升图像分割精度。

技术领域

本申请涉及图像处理技术领域，具体涉及一种图像分割方法及相关装置。

背景技术

语义分割的本质是将图像中每一个像素按照其所属类别进行分类，即将图像按目标类别分割成不同的像素区域。语义分割作为场景理解的一个重要组成部分，在目标识别、图像检索、交通违法识别及自动驾驶等领域有着广泛的应用，以交通违法识别为例，现有的语义分割网络通常首先提取输入图像不同层次的特征，然后通过某种方法融合提取到的不同层次的图像特征，然而这些方法并没有考虑到不同层次特征的平衡问题，没有充分的利用所提取的不同层次的特征，因此，降低了图像分割精度。

发明内容

本申请实施例提供了一种图像分割方法及相关装置，可以提升图像分割精度。

第一方面，本申请实施例提供一种图像分割方法，应用于电子设备，所述电子设备预先配置图像分割网络模型，所述图像分割网络模型包括编码模块、特征平衡模块和解码模块，所述方法包括：

将输入图像输入到所述编码模块进行特征提取，得到特征图像，并对所述特征图像逐步进行下采样操作，得到N层图像特征，所述输入图像为交通场景图像，所述N为大于1的整数；

将所述N层图像特征输入到所述特征平衡模块，得到融合后的N个融合图像特征，所述特征平衡模块用于融合不同层次的图像特征；

将所述N个融合图像特征输入到所述解码模块进行上采样，得到N个上采样图像，每一特征融合图像对应一个上采样图像；

将所述N个上采样图像与所述N个融合图像特征进行拼接融合，得到图像分割结果。

第二方面，本申请实施例提供了一种图像分割装置，应用于电子设备，所述电子设备预先配置图像分割网络模型，所述图像分割网络模型包括编码模块、特征平衡模块和解码模块，所述装置包括：

编码单元，用于将输入图像输入到所述编码模块进行特征提取，得到特征图像，并对所述特征图像逐步进行下采样操作，得到N层图像特征，所述输入图像为交通场景图像，所述N为大于1的整数；

特征平衡单元，用于将所述N层图像特征输入到所述特征平衡模块，得到融合后的N个融合图像特征，所述特征平衡模块用于融合不同层次的图像特征；

解码单元，用于将所述N个融合图像特征输入到所述解码模块进行上采样，得到N个上采样图像，每一特征融合图像对应一个上采样图像；

拼接融合单元，用于将所述N个上采样图像与所述N个融合图像特征进行拼接融合，得到图像分割结果。

第三方面，本申请实施例提供一种电子设备，包括处理器、存储器、通信接口以及一个或多个程序，其中，上述一个或多个程序被存储在上述存储器中，并且被配置由上述处理器执行，上述程序包括用于执行本申请实施例第一方面中的步骤的指令。

第四方面，本申请实施例提供了一种计算机可读存储介质，其中，上述计算机可读存储介质存储用于电子数据交换的计算机程序，其中，上述计算机程序使得计算机执行如本申请实施例第一方面中所描述的部分或全部步骤。