UDBE: Unsupervised Diffusion-based Brightness Enhancement in Underwater Images

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Schein, Tatiana Taís, de Almeira, Gustavo Pereira, Brião, Stephanie Loi, de Bem, Rodrigo Andrade, de Oliveira, Felipe Gomes, Drews-Jr, Paulo L. J.
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916585521807360
author Schein, Tatiana Taís
de Almeira, Gustavo Pereira
Brião, Stephanie Loi
de Bem, Rodrigo Andrade
de Oliveira, Felipe Gomes
Drews-Jr, Paulo L. J.
author_facet Schein, Tatiana Taís
de Almeira, Gustavo Pereira
Brião, Stephanie Loi
de Bem, Rodrigo Andrade
de Oliveira, Felipe Gomes
Drews-Jr, Paulo L. J.
contents Activities in underwater environments are paramount in several scenarios, which drives the continuous development of underwater image enhancement techniques. A major challenge in this domain is the depth at which images are captured, with increasing depth resulting in a darker environment. Most existing methods for underwater image enhancement focus on noise removal and color adjustment, with few works dedicated to brightness enhancement. This work introduces a novel unsupervised learning approach to underwater image enhancement using a diffusion model. Our method, called UDBE, is based on conditional diffusion to maintain the brightness details of the unpaired input images. The input image is combined with a color map and a Signal-Noise Relation map (SNR) to ensure stable training and prevent color distortion in the output images. The results demonstrate that our approach achieves an impressive accuracy rate in the datasets UIEB, SUIM and RUIE, well-established underwater image benchmarks. Additionally, the experiments validate the robustness of our approach, regarding the image quality metrics PSNR, SSIM, UIQM, and UISM, indicating the good performance of the brightness enhancement process. The source code is available here: https://github.com/gusanagy/UDBE.
format Preprint
id arxiv_https___arxiv_org_abs_2501_16211
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle UDBE: Unsupervised Diffusion-based Brightness Enhancement in Underwater Images
Schein, Tatiana Taís
de Almeira, Gustavo Pereira
Brião, Stephanie Loi
de Bem, Rodrigo Andrade
de Oliveira, Felipe Gomes
Drews-Jr, Paulo L. J.
Computer Vision and Pattern Recognition
Artificial Intelligence
Image and Video Processing
Activities in underwater environments are paramount in several scenarios, which drives the continuous development of underwater image enhancement techniques. A major challenge in this domain is the depth at which images are captured, with increasing depth resulting in a darker environment. Most existing methods for underwater image enhancement focus on noise removal and color adjustment, with few works dedicated to brightness enhancement. This work introduces a novel unsupervised learning approach to underwater image enhancement using a diffusion model. Our method, called UDBE, is based on conditional diffusion to maintain the brightness details of the unpaired input images. The input image is combined with a color map and a Signal-Noise Relation map (SNR) to ensure stable training and prevent color distortion in the output images. The results demonstrate that our approach achieves an impressive accuracy rate in the datasets UIEB, SUIM and RUIE, well-established underwater image benchmarks. Additionally, the experiments validate the robustness of our approach, regarding the image quality metrics PSNR, SSIM, UIQM, and UISM, indicating the good performance of the brightness enhancement process. The source code is available here: https://github.com/gusanagy/UDBE.
title UDBE: Unsupervised Diffusion-based Brightness Enhancement in Underwater Images
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Image and Video Processing
url https://arxiv.org/abs/2501.16211