[发明专利]图像分割网络的训练方法、图像分割方法和存储介质

权利要求书

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

将第一分辨率图像输入第一待训练图像分割网络中，得到第一分割结果；所述第一分辨率图像为对第二分辨率图像进行图像映射所得到的，所述第一分辨率图像的分辨率低于所述第二分辨率图像的分辨率；

将所述第二分辨率图像和所述第一分割结果输入第二待训练图像分割网络中，得到第二分割结果；

根据所述第二分割结果和所述第二分辨率图像对应的分割标注数据，调整所述第一待训练图像分割网络和所述第二待训练图像分割网络的网络参数，以得到训练收敛的第一图像分割网络。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

设置所述第二待训练图像分割网络的学习率和所述第一待训练图像分割网络的学习率；其中所述第二待训练图像分割网络的学习率小于所述第一待训练图像分割网络的学习率，使得所述第二待训练图像分割网络的收敛速度与所述第一待训练图像分割网络的收敛速度匹配。

3.根据权利要求1所述的方法，其特征在于，所述根据所述第二分割结果和所述第二分辨率图像对应的分割标注数据，调整所述第一待训练图像分割网络和所述第二待训练图像分割网络的网络参数，包括：

计算所述第二分割结果和所述分割标注数据之间的第一损失，根据所述第一损失调整所述第一待训练图像分割网络和所述第二待训练图像分割网络的网络参数。

4.根据权利要求3所述的方法，其特征在于，所述方法还包括：

根据预设规则，计算所述第一分割结果和所述分割标注数据之间的第二损失；所述预设规则用于表征所述第二分辨率图像的像素点中，与第一分辨率图像中当前像素点所对应的像素点的分割标注数据分布；

相应的，所述根据所述第一损失调整所述第一待训练图像分割网络和所述第二待训练图像分割网络的网络参数，包括：

根据所述第一损失和所述第二损失，调整所述第一待训练图像分割网络和所述第二待训练图像分割网络的网络参数。

5.根据权利要求1所述的方法，其特征在于，所述将所述第二分辨率图像和所述第一分割结果输入第二待训练图像分割网络中，得到第二分割结果，包括：

将所述第一分割结果输入所述第二待训练图像分割网络的上采样层，通过所述上采样层将所述第一分割结果上采样至与所述第二分辨率图像尺寸相同的空间域上；

将所述第二分辨率图像输入所述第二待训练图像分割网络的第一卷积层，通过所述第一卷积层对所述第二分辨率图像中的像素点特征进行提取，得到第一特征图；

将上采样后的第一分割结果和所述第一特征图通道连接后输入所述第二待训练图像分割网络的第二卷积层，得到所述第二分割结果。

6.根据权利要求1所述的方法，其特征在于，所述将所述第二分辨率图像和所述第一分割结果输入第二待训练图像分割网络中，得到第二分割结果，包括：

将所述第一分辨率图像、所述第二分辨率图像和所述第一分割结果输入所述第二待训练图像分割网络中，得到第二分割结果。

7.根据权利要求6所述的方法，其特征在于，所述将所述第一分辨率图像、所述第二分辨率图像和所述第一分割结果输入所述第二待训练图像分割网络中，得到第二分割结果，包括：

将所述第一分辨率图像和所述第一分割结果分别输入所述第二待训练图像分割网络的上采样层，通过所述上采样层将所述第一分辨率图像上采样至与所述第二分辨率图像尺寸相同的空间域上、以及将所述第一分割结果上采样至与所述第二分辨率图像尺寸相同的空间域上；

将上采样后的第一分辨率图像和所述第二分辨率图像通道连接后输入所述第二待训练图像分割网络的第三卷积层，通过所述第三卷积层对上采样后的第一分辨率图像中的像素点特征和所述第二分辨率图像中的像素点特征进行提取融合，得到第二特征图；

将所述第二特征图和上采样后的第一分割结果通道连接后输入所述第二待训练图像分割网络的第四卷积层，得到所述第二分割结果。