SCI躺平式涨点!3 行代码插入Scharr算子注意力,性能直接碾压基线!| 模型改进【V1创新 017】
基于Scharr算子的边缘特征增强模块:
作用位置:任何单一输出特征后,或者任何即插即用模块中。
主要功能:通过Scharr边缘检测算子提取特征图边缘信息,并生成注意力权重来增强原始特征。
代码层面:①Scharr卷积层:使用3×3 Scharr算子计算水平和垂直方向边缘。
②边缘强度计算:通过L2范数融合双方向边缘响应。
③归一化与激活:对边缘图进行批归一化(BN)和GELU激活。
④注意力机制:将归一化的边缘图作为注意力掩码与输入特征相乘。
立即观看