一种基于CBAMT的轴承表面特征的权重调整方法、系统和存储介质

本发明的目的在于提供一种基于CBAMT的轴承表面特征的权重调整方法、系统和存储介质，首先，对于编码阶段经过初步卷积处理的第一特征图像，先通过通道注意力机制强化缺陷相关通道，再分析强化后特征图内像素点的空间相关性，生成第一空间权重特征图；然后，将编码阶段处于跳跃连接时的第二特征图像与第一空间权重特征图相乘融合，再对融合后的特征图进行激活和卷积压缩，生成单通道注意力权重图；最后，将权重图与原特征图逐位相乘，抑制背景噪声，显著增强缺陷区域特征响应。