Underwater image enhancement through depth estimation based on random forest

Shen Chuan Tai, Ting Chou Tsai, Jyun Han Huang

Research output: Contribution to journalArticlepeer-review

3 Citations (Scopus)


Light absorption and scattering in underwater environments can result in low-contrast images with a distinct color cast. This paper proposes a systematic framework for the enhancement of underwater images. Light transmission is estimated using the random forest algorithm. RGB values, luminance, color difference, blurriness, and the dark channel are treated as features in training and estimation. Transmission is calculated using an ensemble machine learning algorithm to deal with a variety of conditions encountered in underwater environments. A color compensation and contrast enhancement algorithm based on depth information was also developed with the aim of improving the visual quality of underwater images. Experimental results demonstrate that the proposed scheme outperforms existing methods with regard to subjective visual quality as well as objective measurements.

Original languageEnglish
Article number063026
JournalJournal of Electronic Imaging
Issue number6
Publication statusPublished - 2017 Nov 1

All Science Journal Classification (ASJC) codes

  • Atomic and Molecular Physics, and Optics
  • Computer Science Applications
  • Electrical and Electronic Engineering


Dive into the research topics of 'Underwater image enhancement through depth estimation based on random forest'. Together they form a unique fingerprint.

Cite this