基于边缘特征注意力融合的水下图像增强方法及系统

本发明公开了一种基于边缘特征注意力融合的水下图像增强方法及系统，包括水下图像的多层次特征提取；通过边缘检测工具提取水下图像的边缘信息，并通过注意力机制将边缘信息与多层次特征进行融合，得到重新分配后的加权特征；将多层次特征和加权特征进行融合，生成增强后的水下图像；通过比较增强后的水下图像与水下图像，计算二者的视觉差异损失和对比度损失；利用损失函数对水下图像增强模型进行训练，得到基于边缘特征注意力融合的水下图像增强模型。本发明通过结合感知优化、结构优化和异常优化提升图像增强效果，感知优化提升图像的语义和视觉感知质量，结构优化增强图像的全局结构信息和局部对比度，异常优化减少异常值对图像的影响。