一种深度模型的空间注意力模块及训练方法

    公开(公告)号:CN119863684A

    公开(公告)日:2025-04-22

    申请号:CN202411911005.X

    申请日:2024-12-24

    Abstract: 本申请公开了一种深度模型的空间注意力模块及训练方法。该深度模型的空间注意力模块包括:空间位置编码单元,用于对输入特征图的高度和宽度上的二维位置进行编码,生成位置编码;初始权重生成单元,用于将位置编码与输入特征图进行融合,通过卷积和非线性变换生成空间注意力的初始权重;权重归一化单元,用于对初始权重进行归一化处理,得到归一化后的权重;空间筛选单元,用于将归一化后的权重作用到输入特征图,对输入特征图中不同位置的特征进行空间筛选。通过本申请,解决了现有技术中因忽略空间位置信息而导致的噪声与目标物体难以区分的技术问题,达到了降低噪声对目标物体检测的干扰,提升目标物体检测的精度的技术效果。

Patent Agency Ranking