基于颜色感知融合注意力和背景光驱离对比学习的水下图像增强方法及装置

本发明公开了一种基于颜色感知融合注意力和背景光驱离对比学习的水下图像增强方法及装置，方法包括：基于设计的颜色提取模块提取水下原始图像中的颜色信息；将所述颜色信息输入局部颜色恢复分支，得到分布合理的颜色权重信息；将所述水下原始图像经过三重注意力分支，实现图像中的细节和模糊信息的恢复，得到细节纹理信息丰富的水下增强图像；将所述分布合理的颜色权重信息与细节纹理信息丰富的水下增强图像输入到全局光恢复分支中，采用对比学习策略来调整水下图像的全局光特性，最终得到恢复出来的水下图像增强结果。本发明可以有效提升水下场景下的增强表现，同时能够消除水下复杂环境对图像的退化影响，真实地还原出图像的色彩与纹理细节。