图像分类方法、装置、电子设备和可读存储介质与流程

技术特征：
1.一种图像分类方法，其特征在于，所述方法包括：获取待分类的目标图像；将所述目标图像输入至预设的第一分类模型，通过所述第一分类模型对所述目标图像进行特征处理，得到所述目标图像中目标物体的分类结果；其中，所述第一分类模型通过采用第二分类模型的输出训练得到，所述第二分类模型通过采用所述第一分类模型的输出更新权重。2.根据权利要求1所述的图像分类方法，其特征在于，在将所述目标图像输入至预设的分类模型之前，所述方法还包括：获取样本数据，所述样本数据包括训练样本集以及验证样本集；根据初始第二分类模型对所述训练样本集的分类结果，对初始第一分类模型进行训练，得到中间第一分类模型；根据所述中间第一分类模型对所述验证样本集的分类结果计算所述初始第二分类模型的第一评价指标，并对所述初始第二分类模型的权重进行更新，得到中间第二分类模型；根据所述中间第二分类模型对所述训练样本集的分类结果，对中间所述第一分类模型进行训练，并根据训练后的中间第一分类模型对所述验证样本集的分类结果，计算所述中间第二分类模型的第二评价指标；根据所述第一评价指标以及所述第二评价指标的确定训练终点，得到第一分类模型和第二分类模型。3.根据权利要求2所述的图像分类方法，其特征在于，所述根据所述中间第一分类模型对所述验证样本集的分类结果对所述初始第二分类模型的权重进行更新，得到所述中间第二分类模型的步骤包括：将所述验证样本集输入至所述初始第二分类模型，得到所述初始第二分类模型对所述验证样本集的分类结果；计算所述初始第二分类模型对所述验证样本集的分类结果与所述中间第一分类模型对所述验证样本集的分类结果的损失；其中，所述中间第一分类模型对所述验证样本集的分类结果为所述验证样本集的标签值；将所述损失反向传播，得到所述初始第二分类模型的第一梯度；根据所述第一梯度对所述初始第二分类模型的权重进行更新，得到中间第二分类模型。4.根据权利要求3所述的图像分类方法，其特征在于，所述根据所述第一梯度对所述初始第二分类模型的权重进行更新，得到中间第二分类模型的步骤包括：对所述第一梯度进行求导，得到第二梯度；对所述第二梯度施加设定邻域内的扰动，得到邻域梯度；根据所述第一梯度以及所述邻域梯度，对所述初始第二分类模型的权重进行更新，得到中间第二分类模型。5.根据权利要求2所述的图像分类方法，其特征在于，所述根据所述中间第一分类模型对所述验证样本集的分类结果计算所述初始第二分类模型的第一评价指标的步骤包括：将所述验证样本集输入至所述初始第二分类模型，得到所述初始第二分类模型对所述验证样本集的分类结果；
将所述初始第二分类模型对所述验证样本集的分类结果以及所述中间第一分类模型对所述验证样本集的分类结果进行比较，得到所述初始第二分类模型的准确率；根据所述初始第二分类模型的准确率，得到所述初始第二分类模型的第一评价指标。6.根据权利要求2所述的图像分类方法，其特征在于，所述根据所述初始第二分类模型对所述训练样本集的分类结果，对所述初始第一分类模型进行训练，得到所述中间第一分类模型的步骤包括：将所述训练样本集输入至所述初始第一分类模型，得到所述初始第一分类模型对所述训练样本集的分类结果；计算所述初始第一分类模型对所述训练样本集的分类结果与所述初始第二分类模型对所述训练样本集的分类结果的损失；其中，所述初始第二分类模型对所述训练样本集的分类结果为所述训练样本集的标签值；判断所述损失是否满足第一预设条件；若未满足，将所述损失反向传播，调整所述初始第一分类模型的权重，以此迭代，直至所述损失满足第一预设条件，得到所述中间第一分类模型。7.根据权利要求2所述的图像分类方法，其特征在于，所述根据训练后的中间第一分类模型对所述验证样本集的分类结果计算所述中间第二分类模型的第二评价指标的步骤包括：将所述验证样本集输入至所述中间第二分类模型，得到所述中间第二分类模型对所述验证样本集的分类结果；将所述中间第二分类模型对所述验证样本集的分类结果以及所述训练后的中间第一分类模型对所述验证样本集的分类结果进行比较，得到所述中间第二分类模型的准确率；根据所述中间第二分类模型的准确率，得到所述中间第二分类模型的第二评价指标。8.根据权利要求2所述的图像分类方法，其特征在于，所述根据所述第一评价指标以及所述第二评价指标的确定训练终点，得到第一分类模型和第二分类模型的步骤包括：计算所述第一评价指标与所述第二评价指标的差值；判断所述差值是否满足第二预设条件；若满足，得到第一分类模型和第二分类模型；若未满足，根据训练后的中间第一分类模型对所述验证样本集的分类结果，对所述中间第二分类模型的权重进行更新，并根据权重更新后的中间第二分类模型对所述训练样本集的分类结果，对训练后的中间第一分类模型继续进行训练，以此迭代，直至第一评价指标与第二评价指标的差值满足所述第二预设条件，得到第一分类模型和第二分类模型。9.根据权利要求2所述的图像分类方法，其特征在于，所述根据初始第二分类模型对所述训练样本集的分类结果，对初始第一分类模型进行训练之前，所述方法还包括：将所述训练样本集输入至预设的原始分类模型，得到所述原始分类模型对所述训练样本集的分类结果；计算所述原始分类模型对所述训练样本集的分类结果与所述训练样本集的原始标签值的损失；判断所述损失是否满足第三预设条件；若未满足，将所述损失反向传播，调整所述原始分类模型的权重，以此迭代，直至所述
损失满足第三预设条件，得到所述初始第二分类模型。10.一种图像分类装置，其特征在于，包括：图像获取模块，用于获取待分类的目标图像；图像分类模块，用于将所述目标图像输入至预设的第一分类模型，通过所述第一分类模型对所述目标图像进行特征处理，得到所述目标图像中目标物体的分类结果；其中，所述第一分类模型通过采用第二分类模型的输出训练得到，所述第二分类模型通过采用所述第一分类模型的输出训练得到。11.一种电子设备，其特征在于，所述电子设备包括存储器、处理器及存储在所述存储器中并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现权利要求1至9任一项所述的图像分类方法。12.一种可读存储介质，其特征在于，所述可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时实现权利要求1至9任一项所述的图像分类方法。

技术总结
本发明实施例提供一种图像分类方法、装置、电子设备和可读存储介质，涉及图像处理技术领域。本发明实施例通过采用第二分类模型的输出训练得到第一分类模型，并通过第一分类模型的输出更新第二分类模型的权重，如此，通过强化循环训练的方式得到第一分类模型，在获取待分类的目标图像后，将目标图像输入至第一分类模型进行特征处理，对目标图像中目标物体进行分类时，能大大提高目标图像中目标物体的分类准确性。类准确性。类准确性。


技术研发人员：杨小平
受保护的技术使用者：顺丰科技有限公司
技术研发日：2021.08.09
技术公布日：2023/2/17