Summary of Publicly Available Underwater Image Enhancement Method
Raise Issue or PR to add more
Recommend Awesome_UIE for more UIE methods and papers.
Sea-thru can be downloaded from Kaggle
| Title | Year | Publication | Code |
|---|---|---|---|
| Enhancing the low quality images using Unsupervised Colour Correction Method | 2010 | SMC | Python |
| Underwater Image Enhancement by Wavelength Compensation and Dehazing | 2011 | TIP | MATLAB |
| Enhancing underwater images and videos by fusion | 2012 | CVPR | MATLAB |
| Transmission Estimation in Underwater Single Images | 2013 | ICCV Workshop | Python |
| Single underwater image enhancement using depth estimation based on blurriness | 2015 | ICIP | MATLAB |
| Automatic Red-Channel Underwater Image Restoration | 2015 | JVCIR | Python |
| Underwater image enhancement by dehazing with minimum information loss and histogram distribution prior | 2016 | TIP | MATLAB |
| Single underwater image restoration by blue-green channels dehazing and red channel correction | 2016 | ICASSP | Python |
| Underwater Image Restoration Based on Image Blurriness and Light Absorption | 2017 | TIP | Python |
| Diving into Haze-Lines: Color Restoration of Underwater Images | 2017 | BMVC | MATLAB |
| Shallow-Water Image Enhancement Using Relative Global Histogram Stretching Based on Adaptive Parameter Acquisition | 2018 | MMM | Python |
| Color Balance and Fusion for Underwater Image Enhancement | 2018 | TIP | MATLAB |
| Enhancement of Underwater Images With Statistical Model of Background Light and Optimization of Transmission Map | 2019 | IEEE T BROADCAST | Python |
| Real-time Model-based Image Color Correction for Underwater Robots | 2019 | IROS | C++ |
| Sea-Thru: A Method for Removing Water From Underwater Images | 2019 | CVPR | Python |
| Restoration of Non-rigidly Distorted Underwater Images using a Combination of Compressive Sensing and Local Polynomial Image Representations | 2019 | ICCV | MATLAB |
| Local Color Mapping combined with Color Transfer for Underwater Image Enhancement | 2019 | WACV | MATLAB |
| Adaptive Enhancement of Underwater Images using Multiobjective PSO | 2019 | MTA | MATLAB |
| L^2UWE: A Framework for the Efficient Enhancement of Low-Light Underwater Images Using Local Contrast and Multi-Scale Fusion | 2020 | CVPR Workshop | MATLAB |
| A novel dark channel prior guided variational framework for underwater image restoration | 2020 | JVCIR | MATLAB |
| Bayesian retinex underwater image enhancement | 2021 | EAAI | MATLAB |
| TEBCF: Real-World Underwater Image Texture Enhancement Model Based on Blurriness and Color Fusion | 2021 | TGRS | MATLAB|Python |
| An Underwater Image Vision Enhancement Algorithm Based on Contour Bougie Morphology | 2021 | TGRS | MATLAB |
| Underwater Image Enhancement by Attenuated Color Channel Correction and Detail Preserved Contrast Enhancement | 2022 | IEEE Journal of Oceanic Engineering | MATLAB |
| Underwater Image Enhancement via Minimal Color Loss and Locally Adaptive Contrast Enhancement | 2022 | TIP | MATLAB |
| Underwater Image Enhancement With Hyper-Laplacian Reflectance Priors | 2022 | TIP | MATLAB |
| A Perception-Aware Decomposition and Fusion Framework for Underwater Image Enhancement | 2022 | TCSVT | MATLAB |
| A Variational Framework for Underwater Image Dehazing and Deblurring | 2022 | TCSVT | MATLAB |
| Underwater Image Enhancement via Piecewise Color Correction and Dual Prior Optimized Contrast Enhancement | 2022 | SPL | MATLAB |
| Towards Underwater Image Restoration: A Physical-accurate Pipeline and a Large Scale Full-reference Benchmark | 2022 | ICME | Python |
| Non-uniform Illumination Underwater Image Restoration via Illumination Channel Sparsity Prior | 2023 | TCSVT | MATLAB |
| Underwater Image Enhancement via Weighted Wavelet Visual Perception Fusion | 2023 | TCSVT | MATLAB |
| Underwater camera: Improving visual perception via adaptive dark pixel prior and color correction | 2023 | IJCV | Python |
| Rank-One Prior: Real-time Scene Recovery | 2023 | TPAMI | MATLAB |
| Underwater Image Enhancement via Piecewise Color Correction and Dual Prior Optimized Contrast Enhancement | 2023 | SPL | MATLAB |
| Enhancing Underwater Image via Adaptive Color and Contrast Enhancement, and Denoising | 2023 | EAAI | MATLAB |
| HFM: A hybrid fusion method for underwater image enhancement | 2024 | EAAI | MATLAB |
| Underwater Color Disparities: Cues for Enhancing Underwater Images Toward Natural Color Consistencies | 2024 | TCSVT | Python |
| Underwater Image Enhancement via Adaptive Color Correction and Stationary Wavelet Detail Enhancement | 2024 | Access | MATLAB |
| CBLA: Color-Balanced Locally Adjustable Underwater Image Enhancement | 2024 | TIM | MATLAB |
| Polarization-Driven Solution for Mitigating Scattering and Uneven Illumination in Underwater Imagery | 2024 | TGRS | MATLAB |
| Name | Year | Publication | Code |
|---|---|---|---|
| WaterGAN | 2017 | RA-L | Tensorflow |
| UGAN | 2018 | ICRA | Tensorflow|Pytorch |
| WAug Encoder-Decoder | 2019 | CVPR Workshop | Pytorch |
| UIE-DAL | 2019 | CVPR Workshops | Pytorch |
| WaterNet | 2019 | TIP | Tensorflow|Pytorch |
| GAN-RS | 2019 | IEEE T IND ELECTRON | Pytorch |
| UResnet | 2019 | Access | Pytorch |
| FUnIE-GAN | 2020 | RA-L | Tensorflow & Pytorch |
| SESR | 2020 | RSS | Tensorflow |
| UWCNN | 2020 | PR | Tensorflow|Pytorch |
| HybridDetectionGAN | 2020 | TCSVT | Tensorflow |
| cGAN | 2020 | SPIC | Tensorflow |
| PRWNet | 2021 | ICCV Workshop | Pytorch |
| CWR | 2021 | IGARSS | Pytorch |
| Shallow-UWNet | 2021 | AAAI | Pytorch |
| UColor | 2021 | TIP | Tensorflow|Pytorch |
| UIEC^2-Net | 2021 | SPIC | Pytorch |
| PFATL | 2022 | IEEE Journal of Oceanic Engineering | Pytorch |
| STSC | 2022 | ICRA | Pytorch |
| URSCT-SESR | 2022 | TGRS | Pytorch |
| TACL | 2022 | TIP | Pytorch |
| TOPAL | 2022 | TCSVT | Pytorch |
| SGUIE-Net | 2022 | TIP | Pytorch |
| PUIE-Net | 2022 | ECCV | Pytorch |
| UIE-WD | 2022 | ICASSP | Pytorch |
| SCNet | 2022 | ICASSP | Pytorch |
| UHD-SFNet | 2022 | ACCV | Pytorch |
| AutoEnhancer | 2022 | ACCV | Pytorch |
| ADMNNet | 2022 | KBS | Pytorch |
| LANet | 2022 | RA-L | Pytorch |
| SIBM | 2022 | MM | Pytorch |
| MBANet | 2022 | PR | Pytorch |
| USUIR | 2022 | AAAI | Pytorch |
| UIR-Net | 2022 | Remote Sensing | Pytorch |
| UDAformer | 2022 | C & G | Pytorch |
| PHYSICAL IMAGE FORMATION MODEL | 2022 | ICASSP | Pytorch |
| UIESS | 2022 | Access | Pytorch |
| U-Shape Transformer | 2023 | TIP | Pytorch |
| PUGAN | 2023 | TIP | Pytorch |
| Semi-UIR | 2023 | CVPR | Pytorch |
| SyreaNet | 2023 | ICRA | Pytorch |
| Five A+ Network | 2023 | BMVC | Pytorch |
| UIALN | 2023 | TCSVT | Pytorch |
| Deep WaveNet | 2023 | TOMM | Pytorch |
| DM | 2023 | MM | Pytorch |
| P2CNet | 2023 | TCSVT | Pytorch |
| TCTL-Net | 2023 | TCSVT | Pytorch |
| CLUIE-Net | 2023 | TCSVT | Pytorch |
| HAAM-GAN | 2023 | EAAI | Pytorch |
| ASNet+ANNet | 2023 | EAAI | Pytorch |
| LiteEnhanceNet | 2023 | ESWA | Pytorch |
| GUPDM | 2023 | MM | Pytorch |
| Boths | 2023 | GRSL | Pytorch |
| TUDA | 2023 | TIP | Pytorch |
| Spectroformer | 2023 | WACV | Pytorch |
| RAUNE-Net | 2023 | IFTC | Pytorch |
| Joint_ID | 2023 | IEEE Sensors | Pytorch |
| UIEPTA | 2023 | IJCNN | Pytorch |
| NU2Net | 2023 | AAAI | Pytorch |
| ACU-Net | 2024 | SIVP | Pytorch |
| UWFormer | 2024 | IJCNN | Pytorch |
| DICAM | 2024 | Neurocomputing | Pytorch |
| INSPIRATION | 2024 | EAAI | Pytorch |
| AoSRNet | 2024 | KBS | Pytorch |
| IACC | 2024 | TGRS | Pytorch |
| TFUIE | 2024 | TGRS | Pytorch |
| DiffWater | 2024 | JSTAR | Pytorch |
| SFGNet | 2024 | ICASSP | Pytorch |
| WsUIDNet | 2024 | TGRS | Pytorch |
| CCMSRNet | 2024 | TGRS | Pytorch |
| SimUESR | 2024 | Neurocomputing | Pytorch |
| MDP | 2024 | IEEE Journal of Oceanic Engineering | Tensorflow |
| CECF | 2024 | AAAI | Pytorch |
| DGD-cGAN | 2024 | PR | Pytorch |
| HCLR-Net | 2024 | IJCV | Pytorch |
| WPFNet | 2024 | KBS | Pytorch |
| GC-UIE | 2024 | KBS | Pytorch |
| DDformer | 2024 | KBS | Pytorch |
torchmetrics for cuda calculation
PSNR
SSIM
MAE
MSE
RMSE
UIQM
UCIQE
NIQE
PS
MUSIQ