图像分类方法及装置

授权

摘要

本发明提供一种图像分类方法及装置，该图像分类方法包括：获取图像数据；将图像数据输入至图像分类模型中，得到图像分类模型输出的类别信息；其中，图像分类模型为基于样本图像数据、样本图像数据的自注意力特征和教师类别信息进行蒸馏训练得到的，样本图像数据的自注意力特征和教师类别信息为教师模型基于样本图像数据得到的，教师模型用于从样本图像数据中提取出自注意力特征教师类别信息，并基于所述自注意力特征进行图像分类。本发明提供的图像分类方法及装置，采用了基于注意力驱动的蒸馏训练方法，能够对图像分类模型进行快速准确地训练，能够节省人工成本，能够避免漏检的问题，能够提高图像分类的准确率，提高图像分类的效率。

基本信息

专利标题：

图像分类方法及装置

专利标题（英）：

暂无

公开（公告）号：

CN113989577A

申请号：

CN202111594126.2

公开（公告）日：

2022-01-28

申请日：

2021-12-24

授权号：

CN113989577B

授权日：

2022-04-05

发明人：

王金桥朱优松李朝闻赵朝阳

申请人：

中科视语（北京）科技有限公司

申请人地址：

北京市门头沟区石龙经济开发区永安路20号3号楼A-6193室

代理机构：

北京路浩知识产权代理有限公司

代理人：

吴刚

优先权：

CN202111594126.2

主分类号：

G06V10/764

IPC分类号：

G06V10/764 G06V10/774 G06V10/82 G06K9/62 G06N3/04 G06N3/08