Underwater Image Enhancement Based on a Spiral Generative Adversarial Framework

Abstract

Underwater image enhancement has drawn much attention due to the significance of underwater vision. Although considerable progress has been made in this field, a key problem remains unsolved: how can we extract and enhance minutiae while trying to remove the noise caused by scattering and attenuation? To address this limitation, we propose a new underwater image enhancement technique with a novel spiral generative adversarial framework, named Spiral-GAN, which can effectivelly recover real-world underwater images with more details, vivid colors and better contrast. For steady training and color correction, we include the pixelwise losses that consist of a mean squared error and an angle error in our objective function. In addition, we design our generator with several deconv-conv blocks to preserve the details from the original distorted images. Furthermore, we present a spiral learning strategy for generalizing the enhancing model to effectively recover the real-world underwater images. Finally, we perform a number of qualitative and quantitative evaluations that suggest that our proposed approach can efficiently enhance the quality of underwater images, which can be further used for underwater object detection.

Keywords

• Underwater image enhancement
• generative adversarial networks
• spiral training