[发明专利]图像分割模型训练、图像分割方法、装置及电子设备

权利要求书

1.一种图像分割模型训练方法，其特征在于，所述方法包括：

获取样本图像集；所述样本图像集中包括：至少一个样本图像，以及，每个所述样本图像对应的标签；所述标签用于表征所述样本图像中待分割出的对象；

基于所述样本图像集中的样本图像和每个所述样本图像对应的标签，对神经网络模型进行N轮训练，得到训练好的图像分割模型；所述神经网络模型包括：轻量级特征提取子模型、特征融合子模型，以及，特征复原子模型，其中，进行第i轮训练包括：

通过所述轻量级特征提取子模型中的M个轻量级特征提取器对所述样本图像进行特征提取，得到M组语义分割特征图；所述N≥1，且所述M≥2；

通过所述特征融合子模型对所述M组语义分割特征图进行特征融合，得到融合特征图；所述融合特征图的分辨率小于所述样本图像的分辨率；

通过所述特征复原子模型中的至少一层转置卷积层，将所述融合特征图的分辨率增大至所述样本图像的分辨率，得到预测分割结果；

根据所述预测分割结果，以及，所述样本图像对应的标签对所述神经网络模型进行第i+1轮训练。

2.根据权利要求1所述的方法，其特征在于，所述轻量级特征提取器包括：至少一层多通道卷积层、第一预设数量的卷积层，以及，第二预设数量的空洞卷积层；所述第一预设数量大于所述第二预设数量；所述第二预设数量的空洞卷积层穿插在所述第一预设数量的卷积层中；

所述通过所述轻量级特征提取子模型中的M个轻量级特征提取器对所述样本图像进行特征提取，得到M组语义分割特征图，包括：

针对第j个所述轻量级特征提取器，通过所述至少一层多通道卷积层从第j-1组语义分割特征图的多个颜色通道进行特征提取，得到子语义分割特征图；所述j大于1且小于或等于所述M；第1个所述轻量级特征提取器的所述至少一层多通道卷积层的输入为所述样本图像；

通过所述卷积层和所述空洞卷积层对所述子语义分割特征图进行特征提取，得到第j组语义分割特征图。

3.根据权利要求1或2所述的方法，其特征在于，所述M等于3，第1组语义分割特征图的分辨率大于第2组语义分割特征图的分辨率，且第2组语义分割特征图的分辨率大于第3组语义分割特征图的分辨率，所述通过所述特征融合子模型对所述M组语义分割特征图进行特征融合，得到融合特征图，包括：

对所述第3组语义分割特征图进行转置卷积，得到所述第3组对应的中间特征图；所述第3组对应的中间特征图的分辨率与所述第2组语义分割特征图的分辨率相同；

对所述第3组对应的中间特征图，以及，所述第2组语义分割特征图进行特征融合，得到第一融合特征图；所述第一融合特征图的分辨率与所述第2组语义分割特征图的分辨率相同；

对所述第一融合特征图进行转置卷积，得到所述分辨率增大后的第一融合特征图；所述分辨率增大后的第一融合特征图的分辨率与所述第1组语义分割特征图的分辨率相同；

对所述分辨率增大后的第一融合特征图，以及，所述第1组语义分割特征图进行特征融合，得到融合特征图。

4.根据权利要求3所述的方法，其特征在于，所述对所述分辨率增大后的第一融合特征图，以及，所述第1组语义分割特征图进行特征融合，得到融合特征图，包括：

对所述分辨率增大后的第一融合特征图，以及，所述第1组语义分割特征图进行特征融合，得到初始融合特征图；

对所述初始融合特征图进行空间维度卷积，得到空间维度改变的初始融合特征图；

对所述空间维度改变的初始融合特征图进行语义特征提取，得到所述空间维度改变的初始融合特征图对应的图像语义特征图；

对所述空间维度改变的初始融合特征图对应的图像语义特征图，进行空间维度卷积以及语义特征提取，得到融合特征图。

5.一种图像分割方法，其特征在于，所述方法包括：

接收目标图像；所述目标图像包括待分割出的目标对象；

将所述目标图像输入至训练好的图像分割模型中，得到所述目标图像对应的图像分割结果；所述训练好的图像分割模型为采用如权利要求1-4任一项所述的方法训练得到的；所述图像分割结果中，所述目标对象对应的像素值，与，除了所述目标对象之外的区域的像素值不同。