Deep Learning Underwater Image Color Correction and Contrast Enhancement Based on Hue Preservation

Chia Hung Yeh, Chih Hsiang Huang, Chu Han Lin

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Underwater Image suffers from serious color distortion and low contrast problems because of complex light propagation in the ocean. In view of computing constraints of underwater vehicles, we propose a high-efficiency deep-learning based framework based on hue preservation. The framework contains three convolutional neural networks for underwater image color restoration. At first, we use the first CNN to convert the input underwater image into the grayscale image. Next, we enhanced the grayscale underwater image by the second CNN. And then, we perform the color correction to the input underwater image by the third CNN. At last, we can obtain the color-corrected image by integrating the outputs of three CNNs based on the hue preservation. In our framework, that CNNs specialize on each work can be able to simplify each architecture of CNNs at most and improve the regression quality to achieve the low computing cost and high effeciency. However, the problem of the underwater CNNs is that the underwater training data is too few and without the corresponding ground truth. Thus, we use the unsupervised learning method CycleGAN to train the underwater CNNs. We design a training method as the combination of three CycleGANs that can train the three CNNs at the same time to share the regression status. This training method may let the three CNNs of our proposed framework support each other to avoid the training overfitting and without constraint. By the proposed framework and training method, our method can process the underwater images with high quality and low computing cost. The experimental results have demonstrated the correct colors and high image quality of the proposed method's results, compared with other related approaches.

Original languageEnglish
Title of host publication2019 IEEE International Underwater Technology Symposium, UT 2019 - Proceedings
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781538641880
DOIs
Publication statusPublished - 2019 Apr
Externally publishedYes
Event2019 IEEE International Underwater Technology Symposium, UT 2019 - Kaohsiung, Taiwan
Duration: 2019 Apr 162019 Apr 19

Publication series

Name2019 IEEE International Underwater Technology Symposium, UT 2019 - Proceedings

Conference

Conference2019 IEEE International Underwater Technology Symposium, UT 2019
Country/TerritoryTaiwan
CityKaohsiung
Period2019/04/162019/04/19

Keywords

  • convolutional neural networks
  • deep learning
  • hue preservation
  • image restoration
  • underwater color correction

ASJC Scopus subject areas

  • Automotive Engineering
  • Ocean Engineering
  • Water Science and Technology

Fingerprint

Dive into the research topics of 'Deep Learning Underwater Image Color Correction and Contrast Enhancement Based on Hue Preservation'. Together they form a unique fingerprint.

Cite this