一种基于模型蒸馏的图像类别识别方法、装置、存储介质及终端专利查询专利号|摘要-满商公司网

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

基本信息

申请号	CN202110499204.4	申请日	-
公开（公告）号	CN113408570A	公开（公告）日	2021-09-17
申请公布号	CN113408570A	申请公布日	2021-09-17
分类号	G06K9/62（2006.01）I;G06N3/02（2006.01）I	分类	计算；推算；计数;
发明人	廖丹萍	申请（专利权）人	浙江视觉智能创新中心有限公司
代理机构	北京辰权知识产权代理有限公司	代理人	李小朋
地址	311215浙江省杭州市萧山区宁围街道钱江世纪公园C区1幢1单元
法律状态	-

摘要

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