NUICNet: Non-Uniform Illumination Correction for Underwater Image Using Fully Convolutional Network

Abstract

Absorption and scattering in aqueous media would attenuate light and make imaging difficult. Therefore, an artificial light source is usually utilized to assist imaging in the deep ocean. However, the artificial light source typically alters the light conditions to a large extent, resulting in the non-uniform illumination of images. To solve this problem, we propose a non-uniform illumination correction algorithm based on a fully convolutional network for underwater images. The proposed algorithm model the original image as the addition of the ideal image and a non-uniform light layer. We replace the traditional pooling layer with dilated convolution to expand the receptive field and achieve higher accuracy in non-uniform illumination recognition. To improve the perception ability of the network effectively, the original image and parameters which pre-trained on the ImageNet are concentrated. The concentrated information is used as input to the network. Due to the color shift and blurred details of the underwater image, we design the novel loss function, which includes three parts of feature loss, smooth loss, and adversarial loss. Moreover, we built a dataset of the underwater image with non-uniform illumination. Experiments show that our method performs better in subjective assessment and objective assessment than some traditional methods.