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