摘要
本发明属于图像处理技术领域,涉及一种基于知识蒸馏的图像特征提取方法及装置;将多个图像样本输入教师模型,得到每个图像样本的第一特征向量;将多个图像样本输入学生模型,得到每个图像样本的第二特征向量;将多个第一特征向量中排序相同的特征值组合,得到m个第一组合向量;将多个第二特征向量中排序相同的特征值组合,得到m个第二组合向量;基于多个图像样本的目标类知识蒸馏损失函数得到第一损失函数,基于多个图像样本的皮尔逊相关系数损失函数得到第二损失函数,基于m个组合向量的皮尔逊相关系数损失函数得到第三损失函数,从而得到图像特征损失函数对学生模型进行训练,直到图像特征提取损失函数最小,得到训练好的学生模型。
技术关键词
皮尔逊相关系数
图像特征提取方法
样本
蒸馏
特征值
学生
特征提取模块
教师
模型训练模块
图像处理技术
组合模块
因子
子模块
尺寸