一种基于知识蒸馏的图像特征提取方法及装置

AITNT
正文
推荐专利
一种基于知识蒸馏的图像特征提取方法及装置
申请号:CN202410720061
申请日期:2024-06-05
公开号:CN118710916A
公开日期:2024-09-27
类型:发明专利
摘要
本发明属于图像处理技术领域,涉及一种基于知识蒸馏的图像特征提取方法及装置;将多个图像样本输入教师模型,得到每个图像样本的第一特征向量;将多个图像样本输入学生模型,得到每个图像样本的第二特征向量;将多个第一特征向量中排序相同的特征值组合,得到m个第一组合向量;将多个第二特征向量中排序相同的特征值组合,得到m个第二组合向量;基于多个图像样本的目标类知识蒸馏损失函数得到第一损失函数,基于多个图像样本的皮尔逊相关系数损失函数得到第二损失函数,基于m个组合向量的皮尔逊相关系数损失函数得到第三损失函数,从而得到图像特征损失函数对学生模型进行训练,直到图像特征提取损失函数最小,得到训练好的学生模型。
技术关键词
皮尔逊相关系数 图像特征提取方法 样本 蒸馏 特征值 学生 特征提取模块 教师 模型训练模块 图像处理技术 组合模块 因子 子模块 尺寸
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号