专利类型:发明专利
发明人:王金桥,朱优松,李朝闻,赵朝阳申请号:CN202111594126.2
申请日:20211224
公开号:CN113989577B
公开日:
20220405
专利内容由知识产权出版社提供
摘要:本发明提供一种图像分类方法及装置,该图像分类方法包括:获取图像数据;将图像数据输入至图像分类模型中,得到图像分类模型输出的类别信息;其中,图像分类模型为基于样本图像数据、样本图像数据的自注意力特征和教师类别信息进行蒸馏训练得到的,样本图像数据的自注意力特征和教师类别信息为教师模型基于样本图像数据得到的,教师模型用于从样本图像数据中提取出自注意力特征教师类别信息,并基于所述自注意力特征进行图像分类。本发明提供的图像分类方法及装置,采用了基于注意力驱动的蒸馏训练方法,能够对图像分类模型进行快速准确地训练,能够节省人工成本,能够避免漏检的问题,能够提高图像分类的准确率,提高图像分类的效率。 申请人:中科视语(北京)科技有限公司
地址:102300 北京市门头沟区石龙经济开发区永安路20号3号楼A-6193室
国籍:CN
代理机构:北京路浩知识产权代理有限公司
代理人:吴刚