基于轻量级注意力机制的水下图像色彩复原方法

    公开(公告)号:CN116563693A

    公开(公告)日:2023-08-08

    申请号:CN202310491245.8

    申请日:2023-05-04

    摘要: 本发明公开了一种基于轻量级注意力机制的水下图像色彩复原方法,包括以下步骤:获取水下图像;构建生成对抗网络模型,包括设计生成器网络结构同时在编码器‑译码器的浅层加入效果较好的轻量级注意力机制,用来更好地提取水下图像的特征信息,以及包括设计鉴别器网络结构以判别生成水下图像的真假;确定用以更新网络参数的基于均方对数误差计算方法的损失函数;使用训练集对设计好的网络结构进行训练,通过循环迭代不断优化网络,直至输出网络模型,并将测试集输入训练好的网络模型,获得清晰图像。通过本发明所增强的水下图像更符合人眼的观察习惯,在峰值信噪比、结构相似性等水下图像画质评测指标方面均相较于现有深度学习方法有所提升。