基于频空跨域Transformer和混合协同表示的水下图像增强方法

本发明涉及一种基于频空跨域Transformer和混合协同表示的水下图像增强方法，包括：获取水下图像组成训练集并进行预处理；基于频空跨域Transformer与混合协同表示，构建水下图像增强模型；将预处理后的水下图像输入水下图像增强模型进行训练；获取待增强水下图像并进行预处理；将预处理后的待增强水下图像输入训练后的水下图像增强模型，获得水下图像增强结果。本发明与利用频空跨域Transformer模块更有效地利用全局内容，将空间域和频率域的信息叠加并放入Transformer的自注意力层中，能够充分利用图像在不同域中的特征，通过自注意力机制捕捉这两种跨域特征，显著的改善图像偏色问题，从而实现对图像的增强。