基于注意力机制和知识蒸馏的图像分类方法

AITNT
正文
推荐专利
基于注意力机制和知识蒸馏的图像分类方法
申请号:CN202410736731
申请日期:2024-06-07
公开号:CN118762248B
公开日期:2025-01-28
类型:发明专利
摘要
本发明提供了一种基于注意力机制和知识蒸馏的图像分类方法,包括:构建教师模型和学生模型,并基于历史训练图像对教师模型进行第一训练;将注意力机制与学生模型进行封装,并基于第一训练结果和历史训练图像对封装结果进行第二训练,得到图像分类预测模型;基于图像分类预测模型对待分类图像进行分析和处理,得到对待分类图像的图像分类结果。通过训练得到的图像分类预测模型对待分类图像进行分析和处理,实现对待分类图像进行准确有效的分类,保障了分类结果的准确性以及可靠性。
技术关键词
分类预测模型 注意力机制 图像分类方法 教师 模型块 学生 蒸馏 图像特征集 负反馈调节 特征生成图像 图像类别标签 层级 参数 管理终端 图像库 工作流
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号