[发明专利]图像分类方法、装置、计算机设备及介质

说明书

本申请实施例公开了一种图像分类方法、装置、计算机设备及介质，属于图像处理领域。方法包括：调用分类模型中的第一分类子模型，对目标图像进行分类处理，得到目标图像的第一预测结果；调用分类模型中的第二分类子模型，对目标图像进行分类处理，得到目标图像的第二预测结果；调用分类模型中的第三分类子模型，对目标图像进行分类处理，得到目标图像的第三预测结果，第三分类子模型的训练样本数据包括样本图像和根据多个标注结果统计得到的结果；输出第三预测结果和目标图像的分类难度，分类难度根据第一预测结果和第二预测结果确定，通过分类难度可以指示模型输出结果的准确性，避免模型对用户的误导。

技术领域

本申请涉及图像处理领域，特别涉及一种图像分类方法、装置、计算机设备及介质。

背景技术

随着人工智能技术和图像处理技术的发展，目前图像的分类处理越来越多地基于分类模型进行。在训练分类模型之后，计算机设备可以调用分类模型对图像进行分类处理，得到该图像的分类结果，以确定该图像是否属于分类模型的目标类别。但是该分类模型仅能输出分类结果，如果分类结果也不够准确，可能会对用户造成误导。

发明内容

本申请实施例提供了一种图像分类方法、装置、计算机设备及介质，可以减少分类模型对用户的误导。所述技术方案如下：

一方面，提供一种图像分类方法，所述方法包括：

调用分类模型中的第一分类子模型，对目标图像进行分类处理，得到所述目标图像的第一预测结果，所述第一分类子模型的训练样本数据包括所述样本图像和从所述样本图像的多个标注结果中选取的其中一个结果，所述多个标注结果中包括第一预设结果或第二预设结果的至少一种，所述多个标注结果中所述第一预设结果的选取可能性大于所述第二预设结果的选取可能性；

调用所述分类模型中的第二分类子模型，对所述目标图像进行分类处理，得到所述目标图像的第二预测结果，所述第二分类子模型的训练样本数据包括所述样本图像和从所述多个标注结果中选取的其中一个结果，所述多个标注结果中所述第二预设结果的选取可能性大于所述第一预设结果的选取可能性；

调用所述分类模型中的第三分类子模型，对所述目标图像进行分类处理，得到所述目标图像的第三预测结果，所述第三分类子模型的训练样本数据包括所述样本图像和根据所述多个标注结果统计得到的结果；

输出所述第三预测结果和所述目标图像的分类难度，所述分类难度根据所述第一预测结果和所述第二预测结果确定。

再一方面，提供了一种图像分类装置，所述装置包括：

第一调用模块，用于调用分类模型中的第一分类子模型，对目标图像进行分类处理，得到所述目标图像的第一预测结果，所述第一分类子模型的训练样本数据包括所述样本图像和从所述样本图像的多个标注结果中选取的其中一个结果，所述多个标注结果中包括第一预设结果或第二预设结果的至少一种，所述多个标注结果中所述第一预设结果的选取可能性大于所述第二预设结果的选取可能性；

第二调用模块，用于调用所述分类模型中的第二分类子模型，对所述目标图像进行分类处理，得到所述目标图像属于所述目标类别的第二预测结果，所述第二分类子模型的训练样本数据包括所述样本图像和从所述多个标注结果中选取的其中一个结果，所述多个标注结果中所述第二预设结果的选取可能性大于所述第一预设结果的选取可能性；

第三调用模块，用于调用所述分类模型中的第三分类子模型，对所述目标图像进行分类处理，得到所述目标图像的第三预测结果，所述第三分类子模型的训练样本数据包括所述样本图像和根据所述多个标注结果统计得到的结果；

输出模块，输出所述第三预测结果和所述目标图像的分类难度，所述分类难度根据所述第一预测结果和所述第二预测结果确定。

可选地，所述输出模块，包括：