基于Transformer和深度估计的水下偏振图像复原方法

本发明公开了一种基于Transformer和深度估计的水下偏振图像复原方法，属于水下图像处理技术领域。方法包括：获取水下偏振图像数据集，将水下偏振图像数据集划分为训练集和测试集；训练基于Transformer和深度估计的水下偏振图像复原网络进行，网络包括编码器和解码器，编码器包括U‑Net模块、多尺度内容引导注意力模块、深度估计网络和多尺度聚合注意力模块；解码器包括串联的两个基于多尺度内容引导注意力的特征融合模块；基于测试集对训练后的基于Transformer和深度估计的水下偏振图像复原网络进行测试；利用通过测试的基于Transformer和深度估计的水下偏振图像复原网络进行图像复原。本发明能够显著改善水下图像的复原效果，增强图像纹理信息，提升图像的清晰度和细节表现。