Unsupervised Underwater Image Enhancement Based on Feature Disentanglement

Liu Yancheng,Dong Zhangwei,Zhu Pengli,Liu Siyuan

DOI: https://doi.org/10.11999/JEIT211517

2022-01-01

Abstract:The absorption and scattering properties of the water medium cause different types of distortion in underwater images, which seriously affects the accuracy and effectiveness of subsequent processing. At present, underwater image enhancement methods with supervised learning rely on synthetic underwater paired image sets for training. However, the supervised learning methods are challenging to apply to practical application scenarios because the synthetic data may not accurately model the underlying physical mechanisms of underwater imaging. An unsupervised underwater image enhancement based on feature disentanglement was proposed. On the one hand, considering the difficulty and high cost of acquiring clear-unclear paired datasets in the same scene, a cycle generative adversarial network was employed to convert the underwater image enhancement problem into a style transfer problem to achieve unsupervised learning. On the other hand, the feature disentanglement method was combined to extract the style features and structure features separately to ensure the structural consistency of the images before and after enhancement. The experimental results show that the method can effectively recover the color and texture details of underwater images in the case of unpaired data training.