Underwater Image Restoration via Constrained Color Compensation and Background Light Color Space-Based Haze-Line Model

The quality of underwater imaging is significantly degraded by light scattering and absorption due to water body and suspended particles. To address the issues of color distortion and contrast degradation, we propose a novel underwater image restoration method based on constrained color compensation and background light color space-based haze-line model. Our method begins by applying the constrained color compensation approach to provide targeted intensity adjustments for attenuated color channels. This process corrects color distortion while simultaneously mitigating the risks of overcompensation and insufficient color saturation. Subsequently, the haze-line model is employed in the transformed background light color space to restore the underwater image. Specifically, the color corrected underwater image is transformed to a novel color space, where the background light intensity serves as the origin. In this color space, all haze lines are identified by grouping pixels with similar color characteristics through the superpixel clustering method. Then, the transmission distribution can be estimated based on the haze-line model. Finally, the scattered light components are removed by applying the underwater descattering model with the estimated transmission distribution to the luminance channel of the color corrected underwater image. Comparative experiments implemented on the UIEB and UCCS underwater image datasets demonstrate the superiority of the proposed method in terms of color correction and contrast enhancement when compared with state-of-the-art underwater image restoration techniques. Our codes are available at https://github.com/MinjieWan/C3HLM.