机器学习模型的训练方法和装置、图像的分类方法和装置制造方法及图纸

本公开涉及一种机器学习模型的训练方法和装置、图像的分类方法和装置，涉及人工智能技术领域。该训练方法包括：利用第一机器学习模型，提取待处理图片的特征向量，根据特征向量确定待处理图片的第一分类结果，待处理图片属于第一数据域或第二数据域；根据特征向量，利用第二机器学习模型，确定待处理图片的第二分类结果，第二分类结果包括待处理图片在第一数据域中的分类结果和在第二数据域中的分类结果；根据第一分类结果和第二分类结果，对第一机器学习模型和第二机器学习模型进行对抗训练，使得第二分类结果的准确率低于阈值，训练好的第一机器学习模型用于图片分类。练好的第一机器学习模型用于图片分类。练好的第一机器学习模型用于图片分类。

【技术实现步骤摘要】
机器学习模型的训练方法和装置、图像的分类方法和装置


[0001]本公开涉及人工智能
，特别涉及一种机器学习模型的训练方法、机器学习模型的训练装置、图像的分类方法、图像的分类装置、电子设备和非易失性计算机可读存储介质。

技术介绍

[0002]图像语义分割作为一种图像的分类方法是计算机视觉领域的一个重要课题，其目的是对于输入图像的每一个像素划分一个独特的语义标签。图像语义分割在许多领域中都有重要应用，如自动驾驶、图像生成等。因此，对用于图像语义分割的机器学习模型进行训练就显得尤为重要。
[0003]在相关技术中，通常依赖于大量有标注的训练数据进行训练；或者利用计算机视觉技术通过渲染生成训练数据进行训练。

技术实现思路

[0004]本公开的专利技术人发现上述相关技术中存在如下问题：获取大量有标注的训练数据成本较高，且生成的训练数据与真实情况有偏差，导致训练效果差。
[0005]鉴于此，本公开提出了一种机器学习模型的训练技术方案，能够降低成本并提高训练效果。
[0006]根据本公开的一些实施例，提供了一种机器学习模型的训练方法，包括：利用第一机器学习模型，提取待处理图片的特征向量，根据所述特征向量确定所述待处理图片的第一分类结果，所述待处理图片属于第一数据域或第二数据域；根据所述特征向量，利用第二机器学习模型，确定所述待处理图片的第二分类结果，所述第二分类结果包括所述待处理图片在第一数据域中的分类结果和在第二数据域中的分类结果；根据所述第一分类结果和所述第二分类结果，对所述第一机器学习模型和所述第二机器学习模型进行对抗训练，使得所述第二分类结果的准确率低于阈值，训练好的第一机器学习模型用于图片分类。
[0007]在一些实施例中，所述根据所述第一分类结果和所述第二分类结果，对所述第一机器学习模型和所述第二机器学习模型进行对抗训练包括：根据所述待处理图片属于的数据域，以及所述第一分类结果，生成第一标注结果；根据所述第二分类结果和所述第一标注结果，计算第一损失函数；根据所述第一损失函数训练所述第二机器学习模型。
[0008]在一些实施例中，所述第一数据域中的图片为人工生成的虚拟图片，所述第二数据域中的图片为实际采集的真实图片。
[0009]在一些实施例中，所述根据所述第一分类结果和所述第二分类结果，对所述第一机器学习模型和所述第二机器学习模型进行对抗训练包括：在所述待处理图片属于第二数据域的情况下，生成所述待处理图片的第二标注结果，所述第二标注结果将所述待处理图片标注为属于第一数据域；根据所述第二分类结果和所述第二标注结果计算第二损失函数；根据所述第二损失函数训练所述第一机器学习模型。
[0010]在一些实施例中，所述根据所述第一分类结果和所述第二分类结果，对所述第一机器学习模型和所述第二机器学习模型进行对抗训练包括：在所述待处理图片属于第一数据域的情况下，根据所述第一分类结果和所述待处理图片的第三标注结果计算第三损失函数，所述第三标注结果用于标注所述待处理图片所属的图像类型；根据所述第三损失函数训练所述第一机器学习模型。
[0011]在一些实施例中，所述根据所述待处理图片属于的数据域，以及所述第一分类结果，生成第一标注结果包括：在所述待处理图片属于的第一数据域的情况下，在所述第一分类结果后面加标识位，生成所述第一标注结果，所述标识位的个数与所有图像类型的个数相同；在所述待处理图片属于的第二数据域的情况下，在所述第一分类结果前面加所述标识位，生成所述第一标注结果，所述标识位的个数与所有图像类型的个数相同。
[0012]在一些实施例中，所述第一分类结果为所述待处理图片中每一个像素属于各图像类型的概率，所述待处理图片在第一数据域中的分类结果为所述待处理图片中每一个像素在第一数据域中属于各图像类型的概率，所述待处理图片在第二数据域中的分类结果为所述待处理图片中每一个像素在第二数据域中属于各图像类型的概率。
[0013]在一些实施例中，所述的训练方法，还包括：利用训练好的第一机器学习模型进行图片分类。
[0014]根据本公开的另一些实施例，提供一种机器学习模型的训练装置，包括：第一确定单元，用于利用第一机器学习模型，提取待处理图片的特征向量，根据所述特征向量确定所述待处理图片的第一分类结果，所述待处理图片属于第一数据域或第二数据域；第二确定单元，用于根据所述特征向量，利用第二机器学习模型，确定所述待处理图片的第二分类结果，所述第二分类结果包括所述待处理图片在第一数据域中的分类结果和在第二数据域中的分类结果；训练单元，用于根据所述第一分类结果和所述第二分类结果，根据所述第一分类结果或所述第二分类结果中的至少一个，对所述第一机器学习模型和所述第二机器学习模型进行对抗训练，使得所述第二分类结果的准确率低于阈值，训练好的第一机器学习模型用于图片分类。
[0015]在一些实施例中，所述训练单元根据所述待处理图片属于的数据域，以及所述第一分类结果，生成第一标注结果；根据所述第二分类结果和所述第一标注结果，计算第一损失函数；根据所述第一损失函数训练所述第二机器学习模型。
[0016]在一些实施例中，所述第一数据域中的图片为人工生成的虚拟图片，所述第二数据域中的图片为实际采集的真实图片。
[0017]在一些实施例中，训练单元在所述待处理图片属于第二数据域的情况下，生成所述待处理图片的第二标注结果，所述第二标注结果将所述待处理图片标注为属于第一数据域；根据所述第二分类结果和所述第二标注结果计算第二损失函数；根据所述第二损失函数训练所述第一机器学习模型。
[0018]在一些实施例中，训练单元在所述待处理图片属于第一数据域的情况下，根据所述第一分类结果和所述待处理图片的第三标注结果计算第三损失函数，所述第三标注结果用于标注所述待处理图片所属的图像类型；根据所述第三损失函数训练所述第一机器学习模型。
[0019]在一些实施例中，训练单元在所述待处理图片属于的第一数据域的情况下，在所
述第一分类结果后面加标识位，生成所述第一标注结果，所述标识位的个数与所有图像类型的个数相同；在所述待处理图片属于的第二数据域的情况下，在所述第一分类结果前面加所述标识位，生成所述第一标注结果，所述标识位的个数与所有图像类型的个数相同。
[0020]在一些实施例中，所述第一分类结果为所述待处理图片中每一个像素属于各图像类型的概率，所述待处理图片在第一数据域中的分类结果为所述待处理图片中每一个像素在第一数据域中属于各图像类型的概率，所述待处理图片在第二数据域中的分类结果为所述待处理图片中每一个像素在第二数据域中属于各图像类型的概率。
[0021]在一些实施例中，所述第一确定单元利用训练好的第一机器学习模型进行图片分类。
[0022]根据本公开的又一些实施例，提供一种图像的分类方法，包括：获取待处理图片；利用第一机器学习模型，对待处理图片进行分类，第一机器学习模型根据上述任一实施例所述的训练方法训练得到。
[0023]根据本公开的再一些实本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种机器学习模型的训练方法，包括：利用第一机器学习模型，提取待处理图片的特征向量，根据所述特征向量确定所述待处理图片的第一分类结果，所述待处理图片属于第一数据域或第二数据域；根据所述特征向量，利用第二机器学习模型，确定所述待处理图片的第二分类结果，所述第二分类结果包括所述待处理图片在第一数据域中的分类结果和在第二数据域中的分类结果；根据所述第一分类结果和所述第二分类结果，对所述第一机器学习模型和所述第二机器学习模型进行对抗训练，使得所述第二分类结果的准确率低于阈值，训练好的第一机器学习模型用于图片分类。2.根据权利要求1所述的训练方法，其中，所述根据所述第一分类结果和所述第二分类结果，对所述第一机器学习模型和所述第二机器学习模型进行对抗训练包括：根据所述待处理图片属于的数据域，以及所述第一分类结果，生成第一标注结果；根据所述第二分类结果和所述第一标注结果，计算第一损失函数；根据所述第一损失函数训练所述第二机器学习模型。3.根据权利要求1所述的训练方法，其中，所述第一数据域中的图片为人工生成的虚拟图片，所述第二数据域中的图片为实际采集的真实图片。4.根据权利要求3所述的训练方法，其中，所述根据所述第一分类结果和所述第二分类结果，对所述第一机器学习模型和所述第二机器学习模型进行对抗训练包括：在所述待处理图片属于第二数据域的情况下，生成所述待处理图片的第二标注结果，所述第二标注结果将所述待处理图片标注为属于第一数据域；根据所述第二分类结果和所述第二标注结果计算第二损失函数；根据所述第二损失函数训练所述第一机器学习模型。5.根据权利要求3所述的训练方法，其中，所述根据所述第一分类结果和所述第二分类结果，对所述第一机器学习模型和所述第二机器学习模型进行对抗训练包括：在所述待处理图片属于第一数据域的情况下，根据所述第一分类结果和所述待处理图片的第三标注结果计算第三损失函数，所述第三标注结果用于标注所述待处理图片所属的图像类型；根据所述第三损失函数训练所述第一机器学习模型。6.根据权利要求2所述的训练方法，其中，所述根据所述待处理图片属于的数据域，以及所述第一分类结果，生成第一标注结果包括：在所述待处理图片属于的第一数据域的情况下，在所述第一分类结果后面加标识位，生成所述第一标注结果，所述标识位的个数与所有图像类...

【专利技术属性】
技术研发人员：申童，张炜，梅涛，
申请(专利权)人：北京京东世纪贸易有限公司，
类型：发明
国别省市：