Fusion-based underwater image enhancement with category-specific color correction and dehazing

Underwater imaging is usually affected by water scattering and absorption, resulting in image blur and color distortion. In order to achieve color correction and dehazing for different underwater scenes, in this paper we report a fusion-based underwater image enhancement technique. First, statistics of the hue channel of underwater images are used to divide the underwater images into two categories: color-distorted images and non-distorted images. Then, category-specific combinations of color compensation and color constancy algorithms are used to remove the color shift. Second, a ground-dehazing algorithm using haze-line prior is employed to remove the haze in the underwater image. Finally, a channel-wise fusion method based on the CIE L* a* b* color space is used to fuse the color-corrected image and dehazed image. For experimental validation, we built a setup to acquire underwater images. The experimental results validate that the category-specific color correction strategy is robust to different categories of underwater images and the fusion strategy simultaneously removes haze and corrects color casts. The quantitative metrics on the UIEBD and EUVP datasets validate its state-of-the-art performance.