基于全变分和颜色平衡的水下图像复原  

Underwater Image Restoration Based on Total Variation and Color Balance

作　　者：郝雅丽 侯国家 李月梅 黄宝香 潘振宽 Hao Yali;Hou Guojia;Li Yuemei;Huang Baoxiang;Pan Zhenkuan(College of Computer Science&Technology,Qingdao University,Qingdao 266071,Shandong,China)

机构地区：[1]青岛大学计算机科学技术学院,山东青岛266071

出　　处：《激光与光电子学进展》2023年第16期129-136,共8页Laser & Optoelectronics Progress

基　　金：国家自然科学基金(61901240,42276203);山东省自然科学基金(ZR2019BF042,ZR2021MD001);山东省高等学校青创人才引育计划。

摘　　要：针对水下图像存在的雾化、模糊和颜色失真问题,提出一种基于全变分和颜色平衡的水下图像复原方法。以完整水下光学成像模型为基础,分别结合四叉树细分法与光在水中传播特性估计背景光,利用水下中值暗通道先验估计透射率,并采用共轭梯度和迭代最小二乘法估计模糊核。为提高计算效率,引入交替方向乘子法对变分能量方程进行逆求解得到去雾、去模糊的图像。在此基础上,在YCbCr空间采用颜色平衡算法对颜色通道进行补偿以校正色彩失真。与6种流行的水下图像增强和复原方法进行比较,实验结果表明,所提方法可以有效地去除雾化和模糊、校正色偏、恢复出清晰、色彩真实的水下图像。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.

关 键 词：图像处理 水下图像复原 水下光学成像模型 全变分模型 颜色平衡 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]