Underwater image processing method based on double attention mechanism

The invention discloses an underwater image processing method based on a double attention mechanism. The double attention mechanism (a non-local attention mechanism and a channel attention mechanism)is used for processing an underwater image. In the shallow network, non-local noise suppression and shallow color feature extraction are adopted; in the deep network, detail features are enhanced andcolors of the underwater image are corrected by using channel attention; only one set of end-to-end network model parameters is needed to carry out denoising, enhancement, color correction and other processing on the underwater image, the efficiency is high, and the effect is good. 本发明公开一种于双重注意机制的水下图像处理方法，利用双重注意机制（非局部注意机制和通道注意机制）对水下图像进行处理。在浅层网络，采用非局部抑制噪声和提取浅层颜色特征；在深层网络，利用通道注意来增强水下图像的细节特征和校正颜色；只需一套端到端的网络模型参数即可对水下图象进行去噪、增强和校正颜色等处理，不仅效率高，而且效果好。