图像分割模型的训练方法及装置、存储介质、电子装置与流程

本申请涉及图像处理领域，具体而言，涉及一种图像分割模型的训练方法及装置、存储介质、电子装置。

背景技术：

1、目前针对图像的分割处理通常是利用图像处理模型进行处理。而线状物体，例如电线、毛线、数据线的分割标注相对于其他障碍物的分割标注具有狭窄以及较为分散的特点，因此，并不利于深度学习分割的算法的收敛，从而使得图像分割模型在训练过程中收敛速度较慢。

2、针对相关技术中，针对线状物的图像分割模型在训练过程中收敛速度慢尚未提出有效的解决方案。

技术实现思路

1、本申请实施例提供了一种线状物图像分割模型的训练方法、机器人、存储介质、电子装置，以至少解决相关技术中针对线状物的图像分割模型在训练过程中收敛速度慢等问题。

2、根据本申请的一个可选实施例，提供了一种图像分割模型的训练方法，包括：按照像素类型对样本图像中的每个像素进行标注，得到样本标注图像，其中，上述像素类型包括背景类和目标对象类；通过神经网络中的第一分支网络对上述样本图像进行背景分割，得到第一分割结果，其中，上述神经网络包括上述图像分割模型，上述第一分支网络为利用第一膨胀图像对图像分割模型进行训练得到的中间模型，上述第一膨胀图像为按照第一膨胀系数对上述样本标注图像进行膨胀操作得到的图像；利用上述第一分割结果和第二膨胀图像对上述图像分割模型进行训练，得到目标图像分割模型，其中，上述第二膨胀图像为按照第二膨胀系数对上述样本标注图像进行膨胀操作得到的图像，上述第二膨胀系数小于上述第一膨胀系数，上述目标图像分割模型对上述样本图像中的上述目标对象类与上述背景类的分割准确率大于预设阈值。

3、可选的，上述利用上述第一分割结果和第二膨胀图像对上述图像分割模型进行训练，包括：将上述第一分割结果作为输入，利用上述第二膨胀图像对上述图像分割模型进行监督训练。

4、可选的，上述利用上述第一分割结果和第二膨胀图像对上述图像分割模型进行训练，包括：在对上述图像分割模型进行监督训练的过程中，根据上述第一分割结果和上述第一膨胀图像，计算上述第一分支网络的损失值；利用上述第一分支网络的损失值更新上述神经网络中的参数。

5、可选的，上述利用上述第一分割结果和第二膨胀图像对上述图像分割模型进行训练，得到目标图像分割模型，包括：通过上述神经网络中的第二分支网络对上述样本图像进行背景分割，得到第二分割结果，其中，上述第二分支网络为利用上述第一分割结果和上述第二膨胀图像对上述图像分割模型进行监督训练得到的中间模型；利用上述第二分割结果和第三膨胀图像对上述图像分割模型进行监督训练，得到上述目标图像分割模型，其中，上述第三膨胀图像为按照第三膨胀系数对上述样本标注图像进行膨胀操作得到的图像，上述第三膨胀系数小于上述第二膨胀系数。

6、可选的，上述利用上述第二分割结果和第三膨胀图像对上述图像分割模型进行监督训练，包括：根据上述第二分割结果和上述第二膨胀图像，计算上述第二分支网络的损失值；利用上述第二分支网络的损失值更新上述神经网络中的参数，得到上述图像分割模型。

7、可选的，上述利用上述第二分割结果和第三膨胀图像对上述图像分割模型进行监督训练，得到目标图像分割模型，还包括：获取上述图像分割模型对上述样本图像中的上述目标对象类与上述背景类的第三分割结果；根据上述样本标注图像和上述第三分割结果计算上述神经网络的损失值；根据上述神经网络的损失值更新上述神经网络中的参数，以得到上述目标图像分割模型。

8、可选的，上述第三膨胀系数为1。

9、根据本申请的另一个实施例，还提供了一种图像分割模型的训练装置，包括：标注单元，用于按照像素类型对样本图像中的每个像素进行标注，得到样本标注图像，其中，上述像素类型包括背景类和目标对象类；分割单元，用于通过神经网络中的第一分支网络对上述样本图像进行背景分割，得到第一分割结果，其中，上述神经网络包括上述图像分割模型，上述第一分支网络为利用第一膨胀图像对图像分割模型进行训练得到的中间模型，上述第一膨胀图像为按照第一膨胀系数对上述样本标注图像进行膨胀操作得到的图像；训练单元，用于利用上述第一分割结果和第二膨胀图像对上述图像分割模型进行训练，得到目标图像分割模型，其中，上述第二膨胀图像为按照第二膨胀系数对上述样本标注图像进行膨胀操作得到的图像，上述第二膨胀系数小于上述第一膨胀系数，上述目标图像分割模型对上述样本图像中的上述目标对象类与上述背景类的分割准确率大于预设阈值。

10、根据本申请的又一个实施例，还提供了一种存储介质，所述存储介质中存储有计算机程序，其中，所述计算机程序被设置为运行时执行上述任一项方法实施例中的步骤。

11、根据本申请的又一个实施例，还提供了一种电子装置，包括存储器和处理器，所述存储器中存储有计算机程序，所述处理器被设置为运行所述计算机程序以执行上述任一项方法实施例中的步骤。

12、通过本申请，按照像素类型对样本图像中的每个像素进行标注，得到样本标注图像，通过神经网络中的第一分支网络对样本图像进行背景分割，得到第一分割结果，神经网络包括图像分割模型，第一分支网络为利用第一膨胀图像对图像分割模型进行训练得到的中间模型，第一膨胀图像为按照第一膨胀系数对样本标注图像进行膨胀操作得到的图像，用第一分割结果和第二膨胀图像对图像分割模型进行训练，得到目标图像分割模型，第二膨胀图像为按照第二膨胀系数对样本标注图像进行膨胀操作得到的图像，第二膨胀系数小于第一膨胀系数，目标图像分割模型对样本图像中的目标对象类与背景类的分割准确率大于预设阈值。基于第一膨胀系数的第一膨胀图像对图像分割模型进行训练得到的第一分割结果，结合第二膨胀图像对图像分割模型再次进行训练，加快图像分割模型在训练过程中的收敛速度，利用多级串联式的膨胀操作进行图像分割网络的训练，从而加快图像分割网络模型的训练收敛。采用上述技术方案，解决了相关技术中针对线状物的图像分割模型在训练过程中收敛速度慢等问题，进而达到提高图像分割模型在训练过程中的收敛速度的效果。

技术特征：

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

2.根据权利要求1所述的方法，其特征在于，所述利用所述第一分割结果和第二膨胀图像对所述图像分割模型进行训练，包括：

3.根据权利要求2所述的方法，其特征在于，所述利用所述第一分割结果和第二膨胀图像对所述图像分割模型进行训练，包括：

4.根据权利要求1所述的方法，其特征在于，所述利用所述第一分割结果和第二膨胀图像对所述图像分割模型进行训练，得到目标图像分割模型，包括：

5.根据权利要求4所述的方法，其特征在于，所述利用所述第二分割结果和第三膨胀图像对所述图像分割模型进行监督训练，包括：

6.根据权利要求5所述的方法，其特征在于，所述利用所述第二分割结果和第三膨胀图像对所述图像分割模型进行监督训练，得到目标图像分割模型，还包括：

7.根据权利要求4至6中任一项所述的方法，其特征在于，所述第三膨胀系数为1。

8.一种图像分割模型的训练装置，其特征在于，包括：

9.一种计算机可读的存储介质，其特征在于，所述存储介质中存储有计算机程序，其中，所述计算机程序被设置为运行时执行所述权利要求1至7任一项中所述的方法。

10.一种电子装置，包括存储器和处理器，其特征在于，所述存储器中存储有计算机程序，所述处理器被设置为运行所述计算机程序以执行所述权利要求1至7任一项中所述的方法。

技术总结
本申请提供了一种图像分割模型的训练方法及装置、存储介质、电子装置，上述方法包括：按照像素类型对样本图像中的每个像素进行标注，得到样本标注图像；通过神经网络中的第一分支网络对样本图像进行背景分割，得到第一分割结果，其中，神经网络包括图像分割模型，第一分支网络为利用第一膨胀图像对图像分割模型进行训练得到的中间模型；利用第一分割结果和第二膨胀图像对图像分割模型进行训练，得到目标图像分割模型，其中，目标图像分割模型对样本图像中的目标对象类与背景类的分割准确率大于预设阈值。采用上述技术方案，解决了相关技术中针对线状物的图像分割模型在训练过程中收敛速度慢等问题。

技术研发人员：梁鹤鸣,朱晨阳,李阳
受保护的技术使用者：追觅创新科技（苏州）有限公司
技术研发日：
技术公布日：2024/1/15