一种基于小波Mamba的水下图像增强方法

本发明涉及一种基于小波Mamba的水下图像增强方法，该方法针对水下图像颜色偏移、对比度低、细节模糊的问题，通过提取水下低质图像的浅层特征并输入多尺度编码网络，利用小波Mamba单元对低频颜色信息和高频细节特征进行联合增强，结合下采样逐层压缩获取全局语义特征，完成颜色偏移校正与对比度增强的全局建模；再通过上采样恢复空间分辨率，结合跳跃连接与小波Mamba单元实现纹理细节和颜色信息的重建增强；最终将特征映射至图像域并与输入图像残差融合，生成色彩自然、纹理清晰、对比度均衡的增强水下图像；本发明通过小波Mamba单元实现频域‑空间域的联合特征建模，显著提升了水下图像的色彩真实性、结构清晰度和细节可感知性。