[发明专利]图像分类方法、装置、电子设备及存储介质

摘要

本申请公开一种图像分类方法、装置、电子设备及存储介质，方法包括：获取图像；将图像输入预先训练的分类模型；接收分类模型对图像的分类结果，并输出分类结果；预先训练分类模型的步骤包括：建立原始分类模型，并获取样本数据集；将样本数据集内的样本数据分批次输入原始分类模型进行训练；接收交叉熵损失函数对每批次样本数据计算的确定标签及否定标签的损失值；对每批次样本数据的确定标签及否定标签的损失值进行平衡，并计算平衡损失值；利用所有的平衡损失值作为总损失值对原始分类模型进行优化，直至确定标签的损失值占比不再降低，得到分类模型；原始分类模型学习时，会向标签为确定存在方向学习，提升对图像进行多类别分类时的分类效果。