Underwater image enhancement via two-level wavelet decomposition maximum brightness color restoration and edge refinement histogram stretching

Underwater images suffer color distortions and low contrast. This is because the light is absorbed and scattered when it travels through water. Different underwater scenes result in different color deviations and levels of detail loss in underwater images. To address these issues of color distortion and low contrast, an underwater image enhancement method that includes two-level wavelet decomposition maximum brightness color restoration, and edge refinement histogram stretching is proposed. First, according to the Jaffe-McGlamery underwater optical imaging model, the proportions of the maximum bright channel were obtained to correct the color of underwater images. Then, edge refinement histogram stretching was designed, and edge refinement and denoising processing were performed while stretching the histogram to enhance contrast and noise removal. Finally, wavelet two-level decomposition of the color-corrected and contrast-stretched underwater images was performed, and the decomposed components in equal proportions were fused. The proposed method can restore the color and detail and enhance the contrast of the underwater image. Extensive experiments demonstrated that the proposed method achieves superior performance against state-of-the-art methods in visual quality and quantitative metrics.