基于色彩平衡及校正的水下图像增强算法

Underwater image enhancement based on color balance and correction

针对水下环境中采集到的图像或视频信息中存在的偏色、模糊、色彩不平衡及部分水下环境中存在非正常光照问题,提出一种基于色彩平衡及校正的水下图像增强算法。首先将水下图像或视频使用自动白平衡算法平衡色彩与色调问题,然后利用单尺度Retinex算法与对比度限制自适应直方图均衡(CLAHE)算法去除水下环境中雾状模糊问题与增强对比度,再利用自适应γ校正算法对R,G,B三通道进行亮度调整,最后将R,G,B三通道进行融合,得到经过增强的水下图像或视频。实验结果表明,该算法具有较好的自适应性,有效改善了图像的偏色与模糊问题,使水下图像或视频质量明显提升。适合于人类的视觉特性,同时具有相对较高的图像评价系数。

Aiming at the problems of color deviation,blur,imbalance and abnormal illumination in some underwater environments,an underwater image enhancement algorithm based on color balance and correction is proposed.Firstly,the underwater image or video use automatic white balance algorithm to balance the color and tone problems.Then,the singlescale Retinex algorithm and CLAHE algorithm are used to remove the haze problem and enhance the contrast in underwater environment.Then,the adaptive gamma correction algorithm is used to adjust the brightness of R,G,B three channels.Finally,the R,G,B channels are fused to get the enhanced underwater image or video.Experimental results show that the algorithm has the visual characteristics suitable for human,and has relatively high image evaluation coefficients.The algorithm has good adaptability,effectively improves the color deviation and blur of the image,and improves the underwater image or video quality significantly.