一种基于因果干涉注意力的图像识别方法

    公开(公告)号:CN115170876A

    公开(公告)日:2022-10-11

    申请号:CN202210766486.4

    申请日:2022-06-30

    Inventor: 李革 柳儒杨 李宏

    Abstract: 一种基于因果干涉注意力的图像识别方法,包括以下步骤:S1.数据预处理:将作为训练数据的图像处理为模型需要的大小和格式并进行增强;S2.预处理后的训练数据进入主干基准网络,得到图像的特征图;S3.将特征图在通道维度切分为四份,视为对特征图进行了因果干涉;S4.在每一份特征图中,实现类别‑空间注意力;S5.在每一份的类别‑空间注意力的基础上上,实现点乘注意力;S6.将经过两层注意力模块之后的四个特征图在通道维度上拼接在一起,得到最终特征;S6.将最终特征输入二元分类器,得到预测结果。本发明方法能去除上下文偏移对图像识别带来的干扰,从而提升图像识别的准确率和鲁棒性,解决了现有的注意力机制加重上下文偏移的问题。

Patent Agency Ranking