[发明专利]分割网络训练方法、使用方法、装置、设备及存储介质

说明书

本申请涉及人工智能技术领域，公开了一种分割网络训练方法、装置、设备及存储介质，其中，方法包括：根据第一图片训练集对第一分割网络进行有监督训练，得到初步图像分割网络；利用初步图像分割网络对第一图片训练集及第二图片训练集中的图片进行特征提取，得到对应的第一图片特征集及第二图片特征集；根据第一图片特征集及第二图片特征集对第二分割网络进行训练，得到图像来源识别网络；根据第二图片训练集对初步图像分割网络进行无监督迭代训练，并获取输出的分割结果图；根据图像来源识别网络和分割结果图评估初步图像分割网络是否训练完成；当初步图像分割网络训练完成时，输出目标图像分割网络。

技术领域

本申请涉及人工智能的技术领域，尤其涉及一种分割网络训练方法、使用方法、装置、设备及存储介质。

背景技术

图像分割是一项广泛使用的技术，如证件照换底色、电影电视特效、视频会议等，可以使用该技术将场景中的目标人像图像从背景中分割出来，图像分割技术不仅能带来娱乐价值，在某些场景还能保证用户的隐私。但是，现有技术中，因为图像分割领域内公开的图像数据集有限，且与实际中拍摄到的图片差异较大，导致使用公开数据集训练所得到的图像分割网络对实际中拍摄到的图片进行图片分割的效果较差。

发明内容

本申请的主要目的在于提供一种分割网络训练方法、使用方法、装置、设备及存储介质，旨在结合有监督学习以及无监督学习对分割网络进行训练，提高分割网络对日常生活中拍摄到的图片的分割能力。

第一方面，本申请提供一种分割网络训练方法，包括：

获取第一图片训练集、及第二图片训练集，其中，所述第一图片训练集中的第一图片带有语义分割标签，所述第二图片训练集中的第二图片未带有语义分割标签，并且所述第一图片的主元素类别和所述第二图片中的主元素类别相同；

根据所述第一图片训练集对预设的第一分割网络进行训练，得到初步图像分割网络；

利用所述初步图像分割网络对所述第一图片训练集及所述第二图片训练集中的图片进行特征提取，以获取所述第一图片训练集对应的第一图片特征集，及所述第二图片训练集对应的第二图片特征集；

根据所述第一图片特征集、及所述第二图片特征集对预设的第二分割网络进行训练，得到图像来源识别网络；

根据所述第二图片训练集对所述初步图像分割网络进行迭代训练，并获取所述初步图像分割网络输出的分割结果图；

根据所述图像来源识别网络和所述分割结果图评估所述初步图像分割网络是否训练完成；

当所述初步图像分割网络训练完成时，输出目标图像分割网络。

第二方面，本申请还提供一种分割网络训练装置，所述分割网络训练装置包括：

第一训练图片获取模块：用于获取第一图片训练集、及第二图片训练集，其中，所述第一图片训练集中的第一图片带有语义分割标签，所述第二图片训练集中的第二图片未带有语义分割标签，并且所述第一图片的主元素类别和所述第二图片中的主元素类别相同；

第一网络训练模块：用于根据所述第一图片训练集对预设的第一分割网络进行训练，得到初步图像分割网络；

第二训练图片获取模块：用于利用所述初步图像分割网络对所述第一图片训练集及所述第二图片训练集中的图片进行特征提取，以获取所述第一图片训练集对应的第一图片特征集，及所述第二图片训练集对应的第二图片特征集；

第二网络训练模块：用于根据所述第一图片特征集、及所述第二图片特征集对预设的第二分割网络进行训练，得到图像来源识别网络；

第三网络训练模块：用于根据所述第二图片训练集对所述初步图像分割网络进行迭代训练，并获取所述初步图像分割网络输出的分割结果图；