[发明专利]图像分割模型的训练方法和装置、图像分割方法和装置

说明书

本申请涉及一种图像分割模型的训练方法、装置、计算机可读存储介质和计算机设备，所述方法包括：将获取的第一未标注图像输入所述语义分割网络中进行分割，得到第一分割图像；所述第一分割图像中的每个分割区域表示一个类别，一个分割区域中的各个像素点的类别相同；将所述第一分割图像输入所述判别网络中进行判别，得到所述第一分割图像对应的第一置信度图；所述第一置信度图用于表示所述第一分割图像中各个像素点的类别的可信程度；根据所述第一置信度图对所述第一分割图像中各个像素点进行标注，得到第一标注图像；将所述第一标注图像输入所述语义分割网络中，对所述语义分割网络进行训练。本申请提供的方案可以提高图像分割模型训练的准确性。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种图像分割的训练方法和装置、图像分割方法和装置。

背景技术

随着计算机技术的发展，出现了图像分割技术。通过图像分割，可以图像分成若干个特定的、具有独特性质的区域并提出感兴趣目标的技术和过程，是由图像处理到图像分析的关键步骤。

然而，传统的对图像分割模型的训练方法中，往往也将一些低质量的标注对模型进行训练，存在训练不准确的问题。

发明内容

基于此，有必要针对图像分割模型训练不准确的技术问题，提供一种图像分割模型的训练方法、装置、计算机可读存储介质和计算机设备。

一种图像分割模型的训练方法，所述图像分割模型包括语义分割网络和判别网络，包括：

将获取的第一未标注图像输入所述语义分割网络中进行分割，得到第一分割图像；

将所述第一分割图像输入所述判别网络中进行判别，得到所述第一分割图像对应的第一置信度图；

根据所述第一置信度图对所述第一分割图像中各个像素点进行标注，得到第一标注图像；

将所述第一标注图像输入所述语义分割网络中，对所述语义分割网络进行训练。

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

获取待分割图像；

将所述待分割图像输入训练完成的语义分割网络中，得到目标图像；

所述语义分割网络是基于第一标注图像和获取的第一未标注图像进行训练的，所述第一标注图像是根据第一置信度图对所述第一分割图像中各个像素点进行标注得到的，所述第一置信度图是将第一分割图像输入判别网络中进行判别得到的，所述第一分割图像是所述第一未标注图像输入所述语义分割网络中得到的。

一种图像分割模型的训练装置，所述图像分割模型包括语义分割网络和判别网络，所述装置包括：

分割模块，用于将获取的第一未标注图像输入所述语义分割网络中进行分割，得到第一分割图像；

判别模块，用于将所述第一分割图像输入所述判别网络中进行判别，得到所述第一分割图像对应的第一置信度图；

标注模块，用于根据所述第一置信度图对所述第一分割图像中各个像素点进行标注，得到第一标注图像；

训练模块，用于将所述第一标注图像输入所述语义分割网络中，对所述语义分割网络进行训练。

一种图像分割装置，所述装置包括：

待分割图像获取模块，用于获取待分割图像；

分割模块，用于将所述待分割图像输入训练完成的语义分割网络中，得到目标图像；