一种轻量化水下图像增强的方法

本发明涉及水下图像增强领域，具体涉及一种轻量化水下图像增强的方法，包括：构建水下图像数据集；基于网络基础模块与通道注意力模块，构建卷积下采样层与卷积上采样层；搭建水下图像增强网络，构建网络训练使用的融合损失函数L构建水下图像增强网络的融合损失函数；根据融合损失函数，使用水下图像数据集的水下图像数训练水下图像增强网络；把训练后的水下图像增强网络部署到末端应用设备中运行，通过水下图像增强网络对水下图像进行增强，得到增强的水下图像。本发明可以降低了水下图像增强网络参数量和计算复杂度，提升图像特征提取能力，提高水下图像的质量，解决水下图像的关键退化问题，提升末端设备的感知能力和任务执行效率。