试试“频域”!傅里叶变换让模型秒抓全局频率信息,丰富度直接拉满!| 模型改进【V1创新 018】
二维快速傅里叶变换模块:
作用位置:任何单一输出特征后,或者任何即插即用模块中。
主要功能:增强全局频率信息的捕捉能力,卷积操作实现空间域与频域信息融合,提升特征丰富性。
代码层面:①对输入特征图进行归一化,确保数值稳定性。
②二维快速傅里叶变换(FFT)将特征从空间域转换至频域,得到复数形式的频域张量,使用 1×1 卷积进行频域特征变换,并通过 GELU 激活函数引入非线性;
③通过逆快速傅里叶变换(IFFT)将特征从频域转换回空间域,输出增强后的特征图。
立即观看