Underwater image restoration model based on multi-branch gating fusion and restoration method thereof

Abstract

The invention discloses an underwater image restoration model based on multi-branch gating fusion and a restoration method thereof. The underwater image restoration model comprises a multi-branch feature extraction module, a gating fusion module and a reconstruction module which are connected in sequence. The multi-branch feature extraction module is used for extracting image feature information of different scales and different levels of an underwater original image to be restored; the gating fusion module is used for fusing the image feature information of different scales and different levels of the underwater original image to be restored to obtain an underwater low-resolution feature image; and the reconstruction module is used for carrying out image reconstruction on the underwater low-resolution feature image to obtain a high-resolution feature image, namely an underwater restored image. The multi-branch feature extraction module is composed of a plurality of different branch modules used for extracting image feature information of different scales and different levels of the to-be-restored underwater original image, the sizes of feature maps output by the different branch modules are the same, the contrast ratio is effectively improved, and the chromatic aberration is eliminated.

Description

technical field [0001] The invention belongs to the technical field of underwater image restoration, and relates to an underwater image restoration model based on multi-branch gating fusion and a restoration method thereof. Background technique [0002] my country has a vast ocean area, how to effectively develop and utilize marine resources is particularly urgent for alleviating my country's energy crisis and solving the problem of sustainable development. Due to the serious impact of complex underwater conditions on image acquisition, underwater images often have problems such as low contrast, color distortion, and image blur, which seriously affect subsequent processing and further use. Therefore, it is necessary to study effective solutions to improve the visibility, contrast, and color characteristics of underwater images for better visual effects. [0003] Traditional approaches to underwater image restoration can be divided into two categories: image enhancement and ...

AI Technical Summary

Problems solved by technology

[0006] Embodiments of the present invention provide an underwater image restoration model and restoration method based on multi-branch gated fusion, so as to solve the problem of poor restoration effect of the existing deep learning-based underwater image restoration method and blurred defects in the restored underwater image. The problem

Images