-
公开(公告)号:CN115170876A
公开(公告)日:2022-10-11
申请号:CN202210766486.4
申请日:2022-06-30
Applicant: 北京大学深圳研究生院
IPC: G06V10/764 , G06V10/82 , G06N3/04 , G06N3/08
Abstract: 一种基于因果干涉注意力的图像识别方法,包括以下步骤:S1.数据预处理:将作为训练数据的图像处理为模型需要的大小和格式并进行增强;S2.预处理后的训练数据进入主干基准网络,得到图像的特征图;S3.将特征图在通道维度切分为四份,视为对特征图进行了因果干涉;S4.在每一份特征图中,实现类别‑空间注意力;S5.在每一份的类别‑空间注意力的基础上上,实现点乘注意力;S6.将经过两层注意力模块之后的四个特征图在通道维度上拼接在一起,得到最终特征;S6.将最终特征输入二元分类器,得到预测结果。本发明方法能去除上下文偏移对图像识别带来的干扰,从而提升图像识别的准确率和鲁棒性,解决了现有的注意力机制加重上下文偏移的问题。