Skip to content
/ UDnet Public

Adaptive Uncertainty Distribution in Deep Learning for Unsupervised Underwater Image Enhancement

License

Notifications You must be signed in to change notification settings

alzayats/UDnet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Adaptive Uncertainty Distribution in Deep Learning for Unsupervised Underwater Image Enhancement (Paper)

License Framework

The Pytorch Implementation of ''Adaptive Uncertainty Distribution in Deep Learning for Unsupervised Underwater Image Enhancement''.

Introduction

We present a fully trainable framework to enhance underwater images without ground truth. We demonstrate that our proposed model outperforms ten popular underwater image enhancement methods on seven common metrics, in both paired and unpaired settings.

Requirement

In this project, we use Linux host with a single NVidia GeForce RTX 2080 Ti GPU with 11 GB of memory, Python 3.7, Pytorch 1.9.1.

Running

Inference

Download the pretrained model pretrained model.

Check the model and image pathes in Inference.py, and then run:

python Inference.py  

Training

To train the model, you need to download Underwater Dataset dataset.

Check the dataset path in Train.py, and then run:

python Train.py   

Bibtex

If you find UDnet is useful in your research, please cite our paper:

@article{Saleh2022udnet,
    title = {{Adaptive Uncertainty Distribution in Deep Learning for Unsupervised Underwater Image Enhancement}},
    year = {2022},
    author = {Saleh, Alzayat and Sheaves, Marcus and Jerry, Dean and Azghadi, Mostafa Rahimi},
    month = {12},
    url = {https://arxiv.org/abs/2212.08983v1},
    doi = {10.48550/arxiv.2212.08983},
    arxivId = {2212.08983},
    keywords = {Convolutional Neural Net-works, Deep Learning, Index Terms-Computer Vision, Machine Learning, Underwater Image Enhancement, Variational Autoen-coder}
}

License

The code is made available for academic research purpose only. This project is open sourced under MIT license.

Credit

https://github.com/zhenqifu/PUIE-Net
https://github.com/deepxzy/USLN
https://github.com/xueleichen/PyTorch-Underwater-Image-Enhancement

About

Adaptive Uncertainty Distribution in Deep Learning for Unsupervised Underwater Image Enhancement

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages