一种图像分类方法、装置、终端及存储介质

基本信息

申请号 CN202111583881.0 申请日 -
公开(公告)号 CN114219044A 公开(公告)日 2022-03-22
申请公布号 CN114219044A 申请公布日 2022-03-22
分类号 G06K9/62(2022.01)I;G06V10/40(2022.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I;G06V10/764(2022.01)I 分类 计算;推算;计数;
发明人 雷柏英;熊唯;杨亮;于恺威;汪天富 申请(专利权)人 深圳大学
代理机构 深圳市君胜知识产权代理事务所(普通合伙) 代理人 温宏梅
地址 518060广东省深圳市南山区南海大道3688号
法律状态 -

摘要

摘要 本发明公开了一种图像分类方法、装置、终端及存储介质,方法通过获取待分类图像,将待分类图像输入目标分类模型,其中,目标分类模型包括卷积层、注意力层以及分类层;通过卷积层获取待分类图像的局部特征信息,得到第一特征图;通过注意力层对第一特征图进行全局建模,得到第二特征图;通过分类层对第二特征图进行图像分类,得到待分类图像对应的图像类别。本发明中的目标分类模型可以捕捉待分类图像的局部特征信息也可以进行全局建模,因此可以精确地预测出待分类图像对应的图像类别。解决了现有技术中深度卷积神经网络仅具备有捕捉局部上下文信息的能力,而不具备全局建模的能力,导致模型的分类性能不佳的问题。