[发明专利]基于自蒸馏的图像分类方法、系统、存储介质和电子设备

摘要

本发明公开了一种基于自蒸馏的图像分类方法、系统、存储介质和电子设备，包括：基于多个图像样本，对卷积神经网络进行迭代训练，得到第一图像分类模型；将每个图像样本输入第一图像分类模型，得到并将输出特征图分别输入特征处理模块和注意力机制模块，得到并根据原始特征图和注意力特征图进行自蒸馏，得到并根据每个图像样本的自蒸馏损失值，对第一图像分类模型的参数进行优化，直至第二图像分类模型满足预设条件时，将第二图像分类模型确定为目标图像分类模型；将待测图像输入至目标图像分类模型，得到图像分类结果。本发明能够利用模型自身的注意力图进行蒸馏，在有效压缩图像分类模型训练时间的同时，确保了图像分类的效果不受影响。