Laser & Optoelectronics Progress, Volume. 60, Issue 16, 1610007(2023)
Underwater Image Restoration Based on Total Variation and Color Balance
Aiming at solving the problems of fogging, blurring and color distortion in underwater images, an underwater image restoration method is proposed based on total variation and color balance. Depending on the complete underwater optical imaging model, the background light is estimated by combining the quadtree subdivision algorithm and the propagation characteristics of light in water, the transmittance is estimated by using the underwater median dark channel prior, and the fuzzy kernel is estimated by using the conjugate gradient and iterative least square method. In order to improve the computational efficiency, the alternating direction multiplier method is introduced to inverserly solve the variational energy equation to obtain a haze-free and deblurred image. In addition, a color balance algorithm is proposed to compensate color channel in YCbCr space for correcting color distortion. Compared with six popular underwater image enhancement and restoration methods, the experimental results show that the proposed method can effectively remove fog and blur, correct color deviation, and restore a clear and true color underwater image.
Get Citation
Copy Citation Text
Yali Hao, Guojia Hou, Yuemei Li, Baoxiang Huang, Zhenkuan Pan. Underwater Image Restoration Based on Total Variation and Color Balance[J]. Laser & Optoelectronics Progress, 2023, 60(16): 1610007
Category: Image Processing
Received: Sep. 1, 2022
Accepted: Oct. 18, 2022
Published Online: Aug. 15, 2023
The Author Email: Hou Guojia (hgj2015@qdu.edu.cn)