The invention relates to an
underwater image enhancement method based on
wavelet Mama, which aims at the problems of
color shift,
low contrast ratio and fuzzy details of an
underwater image, extracts shallow layer features of an
underwater low-quality image, inputs the shallow layer features into a multi-scale coding network, and performs joint enhancement on low-frequency color information and high-frequency detail features by using a
wavelet Mama unit. Global semantic features are obtained by combining down-sampling layer-by-layer compression, and
global modeling of color offset correction and
contrast enhancement is completed; the spatial resolution is recovered through up-sampling, and reconstruction and enhancement of texture details and color information are realized in combination with jump connection and a
wavelet Mama unit; and finally, the features are mapped to an
image domain and fused with input image residual errors, and an enhanced underwater image with natural color, clear texture and balanced contrast is generated. According to the method,
frequency domain-space domain joint
feature modeling is realized through the wavelet Mama unit, and the color authenticity, the structural definition and the detail perceptibility of the underwater image are remarkably improved.