[发明专利]一种基于模型蒸馏的图像类别识别方法、装置、存储介质及终端

摘要

本发明公开了一种基于模型蒸馏的图像类别识别方法、装置、存储介质及终端，该方法包括：获取待分类目标图像并输入预先训练的学生模型中，输出多个类别概率值；其中，预先训练的学生模型基于模型蒸馏法训练生成，模型蒸馏法训练生成是基于预先训练的教师模型内全连接层的不同类别分类映射向量之间的相似度训练生成，不同分类映射向量之间的相似度为不同分类映射向量之间夹角的余弦值；基于多个类别概率值识别待分类目标图像的目标类别。因此，本申请实施例通过采用基于预先训练的教师模型内全连接层的不同类别分类映射向量之间的相似度指导学生模型进行训练，使得学生模型也能学到和教师模型相似区分度的特征，由于学生模型结构简单以及参数少，从而提升了硬件平台的运行速度，进一步提高了图像分类效率。