一种基于双U-net的水下图像增强方法

摘要

本发明属于计算机视觉领域，涉及一种基于双U‑net的水下图像增强方法。本发明基于水下成像物理模型设置了两个U‑net连接卷积网络框架，每个U‑net由编码‑解码网络组成，第一个U‑net前添加注意力机制，通过处理输入图像和其灰度图像，在解码结构每层都输出一个特征信息用于第二个U‑net，并估计传输图像。利用第二个U‑net处理水下图像估计补偿红光后的水下图像latent，在解码结构中连接了第一个U‑net输出的特征信息，保证图像增强过程细节不被丢失。最后根据成像模型latent除以传输图为最终结果。本发明能够有效增强颜色失真严重且雾化严重的水下图像，同时保留图像的细节。