一种带有小波变换和融合注意力机制的水下图像增强方法

本发明涉及图像处理领域，尤其涉及一种带有小波变换和融合注意力机制的水下图像增强方法，本发明获取水下图像数据，分析所述图像数据的显性特征信息并计算特征提取表征参数，以设置标签，构建水下图像增强网络，生成器接收图像数据后生成输出图像，判别器获取生成器的输出图像，基于所述标签判定是否调整矩阵网格数量，以输出对应矩阵网格数量的矩阵，对所述输出图像的局部区域进行判定，确定水下图像增强网络的构建状态，利用训练集的水下图像通过复合损失函数对所述水下图像增强网络进行训练直至收敛。本发明通过利用生成器和判别器构建水下图像增强网络，对退化的水下图像进行增强，提高了图像质量和水下细节的还原度。