• Optics and Precision Engineering
  • Vol. 30, Issue 22, 2939 (2022)
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
    DOI: 10.37188/OPE.20223022.2939 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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.
    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