Optics and Precision Engineering, Volume. 30, Issue 22, 2939(2022)

Coarse-to-fine underwater image enhancement based on multi-level wavelet transform

Guoming YUAN1, Guang YANG2、*, Jinfeng WANG2, Haijun LIU1, and Wei WANG2
Author Affiliations
  • 1Department of Emergency Management, Institute of Disaster Prevention, Sanhe06520, China
  • 2Department of Information Engineering, Institute of Disaster Prevention, Sanhe06501, China.
  • show less

    To correct the color distortion and enhance the details of degraded underwater image, this paper proposes a coarse-to-fine underwater image enhancement method based on multi-level wavelet transform. Firstly, a raw underwater image is decomposed into a low-frequency image and a series of high-frequency images based on the wavelet transform. Secondly, a two-stage underwater enhancement network is proposed, which includes a multi-level wavelet transform sub-network and a refinement sub-network with the proposed second-order Runge-Kutta block. The multi-level wavelet transform sub-network, which estimates preliminary result, contains a low-frequency and a high-frequency branch. Specifically, the low-frequency branch treats the color correction problem as the implicit style transfer problem and introduces the instance normalization and the position normalization into the branch. To ensure an accurate reconstruction, when manipulating low-frequency information, the high-frequency branch calculates the enhanced mask according to the information from both low- and high-frequency images and implements the enhancement by multiplying the progressive up-sampling enhanced mask with the high-frequency images. We implemented the inverse wavelet transform and obtain the preliminary results. Finally, the refinement network was designed to further optimize the preliminary results with the proposed second-order Runge-Kutta block. Experimental results demonstrated that the proposed method outperformed the existing methods in enhancement effect on both synthetic and real images, whilst the Peak Signal-to-Noise Ratio (PSNR) improved by 9%. The proposed method also meets the requirement of underwater vision tasks, such as color correction, details enhancement, and clarity.

    Tools

    Get Citation

    Copy Citation Text

    Guoming YUAN, Guang YANG, Jinfeng WANG, Haijun LIU, Wei WANG. Coarse-to-fine underwater image enhancement based on multi-level wavelet transform[J]. Optics and Precision Engineering, 2022, 30(22): 2939

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Information Sciences

    Received: May. 19, 2022

    Accepted: --

    Published Online: Nov. 28, 2022

    The Author Email: YANG Guang (yangg202205@163.com)

    DOI:10.37188/OPE.20223022.2939

    Topics