公开(公告)号：CN113762251B

公开(公告)日：2024-05-10

申请号：CN202110943944.2

申请日：2021-08-17

Applicant: 慧影医疗科技(北京)股份有限公司

Inventor： 柴象飞 ,  郭娜 ,  张路 ,  刘鹏飞 ,  秦浩峰

IPC: G06V10/25 ,  G06V10/764 ,  G06V10/82 ,  G06N3/0464

Abstract: 本发明公开了一种基于注意力机制的目标分类方法，涉及神经网络技术领域。包括：获取待分类图像的步骤；将待分类图像输入到卷积神经网络，提取待分类图像的特征，通过注意力矩阵块对得到的特征图进行处理，根据处理后的特征图得到至少一个目标分类结果的步骤。本发明适用于图像中目标物的分类，通过加入注意力机制，使用注意力矩阵块对特征图进行处理，抑制与目标分类相关性低的特征表达，激励与目标相关性大的特征表达，从而更有效地利用全局信息对目标类型进行预测，抑制无关信息对分类结果的干扰，提高目标分类的准确性。此外，本发明不需要对目标周围的局部图像做切割操作，因而不存在图像切割相关的问题，处理过程更加简洁，效率更高。