一种基于注意力机制的目标分类方法及系统
基本信息
申请号 | CN202110943944.2 | 申请日 | - |
公开(公告)号 | CN113762251A | 公开(公告)日 | 2021-12-07 |
申请公布号 | CN113762251A | 申请公布日 | 2021-12-07 |
分类号 | G06K9/32(2006.01)I;G06K9/62(2006.01)I;G06N3/04(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 柴象飞;郭娜;张路;刘鹏飞;秦浩峰 | 申请(专利权)人 | 慧影医疗科技(北京)股份有限公司 |
代理机构 | 北京轻创知识产权代理有限公司 | 代理人 | 陈霆雷 |
地址 | 100192 北京市海淀区西小口路66号中关村东升科技园·北领地B-2号楼2层A206 | ||
法律状态 | - |
摘要
摘要 | 本发明公开了一种基于注意力机制的目标分类方法,涉及神经网络技术领域。包括:获取待分类图像的步骤;将待分类图像输入到卷积神经网络,提取待分类图像的特征,通过注意力矩阵块对得到的特征图进行处理,根据处理后的特征图得到至少一个目标分类结果的步骤。本发明适用于图像中目标物的分类,通过加入注意力机制,使用注意力矩阵块对特征图进行处理,抑制与目标分类相关性低的特征表达,激励与目标相关性大的特征表达,从而更有效地利用全局信息对目标类型进行预测,抑制无关信息对分类结果的干扰,提高目标分类的准确性。此外,本发明不需要对目标周围的局部图像做切割操作,因而不存在图像切割相关的问题,处理过程更加简洁,效率更高。 |
