摘要
本发明提供了一种基于注意力机制和知识蒸馏的图像分类方法,包括:构建教师模型和学生模型,并基于历史训练图像对教师模型进行第一训练;将注意力机制与学生模型进行封装,并基于第一训练结果和历史训练图像对封装结果进行第二训练,得到图像分类预测模型;基于图像分类预测模型对待分类图像进行分析和处理,得到对待分类图像的图像分类结果。通过训练得到的图像分类预测模型对待分类图像进行分析和处理,实现对待分类图像进行准确有效的分类,保障了分类结果的准确性以及可靠性。
技术关键词
分类预测模型
注意力机制
图像分类方法
教师
模型块
学生
蒸馏
图像特征集
负反馈调节
特征生成图像
图像类别标签
层级
参数
管理终端
图像库
工作流