一种基于频域分析和视觉Mamba的水下图像增强方法

本发明涉及一种基于频域分析和视觉Mamba的水下图像增强方法，包括：构建多级小波变换网络，该网络为编码器‑解码器结构，其中所述编码器包括空间‑频率特征融合模块和离散小波变换；所述解码器包括空间‑频率特征融合模块和逆离散小波变换；在编码器中离散小波变换将水下图像分解为低频部分和高频部分，低频部分和高频部分由对应的空间‑频率特征融合模块分别基于Mamba和频域卷积进行空间特征和频率特征提取，在解码器按照先低频后高频顺序沿通道进行拼接，再经过逆离散小波变换进行重建；将水下图像数据输入多级小波变换网络，得到增强后的水下图像数据。与现有技术相比，本发明具有提升增强质量、降低复杂度、以及快速增强等优点。