一种基于知识蒸馏的素描图处理方法及其系统
基本信息
申请号 | CN202110908616.9 | 申请日 | - |
公开(公告)号 | CN113360701A | 公开(公告)日 | 2021-09-07 |
申请公布号 | CN113360701A | 申请公布日 | 2021-09-07 |
分类号 | G06F16/583;G06F16/55 | 分类 | 计算;推算;计数; |
发明人 | 徐行;田加林;沈复民;申恒涛 | 申请(专利权)人 | 成都考拉悠然科技有限公司 |
代理机构 | 成都市集智汇华知识产权代理事务所(普通合伙) | 代理人 | 李华;温黎娟 |
地址 | 610095 四川省成都市中国(四川)自由贸易试验区成都高新区天府五街200号4号楼A区10层1001、1002、1003室 | ||
法律状态 | - |
摘要
摘要 | 本发明公开了一种基于知识蒸馏的素描图处理方法及其系统,包括:获取预训练的由图像分类模型构成的教师模型,并基于教师模型提取样本图像的教师特征和教师分类概率;构建能够基于素描图进行图像分类的学生模型,并基于学生模型提取样本图像的第一学生特征和第一学生分类概率,并基于学生模型提取样本图像对应样本素描图的第二学生特征和第二学生分类概率;基于教师特征、教师分类概率、第一学生特征、第一学生分类概率、第二学生特征和第二学生分类概率计算分类损失和蒸馏损失,并基于分类损失和蒸馏损失更新学生模型;学生模型基于待检测素描图和待查询图像生成零样本跨模态检索结果。 |
