https://github.com/CXH-Research/Underwater-Image-Enhancement

Summary of Publicly Available Underwater Image Enhancement Method
https://github.com/CXH-Research/Underwater-Image-Enhancement
image-processing underwater-image-enhancement underwater-image-processing underwater-image-restoration underwater-images underwater-robotics
Last synced: 2 months ago
JSON representation
Summary of Publicly Available Underwater Image Enhancement Method
Host: GitHub
URL: https://github.com/CXH-Research/Underwater-Image-Enhancement
Owner: CXH-Research
License: mit
Created: 2023-11-13T04:16:10.000Z (over 1 year ago)
Default Branch: main
Last Pushed: 2025-02-07T02:43:51.000Z (3 months ago)
Last Synced: 2025-02-07T03:27:04.828Z (3 months ago)
Topics: image-processing, underwater-image-enhancement, underwater-image-processing, underwater-image-restoration, underwater-images, underwater-robotics
Homepage:
Size: 44.9 KB
Stars: 103
Watchers: 4
Forks: 15
Open Issues: 0
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project

README

        # Underwater-Image-Enhancement

Summary of Publicly Available Underwater Image Enhancement Method

**Raise Issue or PR to add more**

Recommend [Awesome_UIE](https://github.com/fansuregrin/Awesome-UIE) [UIE](https://github.com/YuZhao1999/UIE) for more UIE methods and papers.

## Dataset

[Github Summary](https://github.com/xinzhichao/underwater_datasets)

[Github Summary](https://github.com/lizhh268/awesome_underwater_image_enhancement-UIE-)

Sea-thru can be downloaded from [Kaggle](https://www.kaggle.com/datasets/colorlabeilat/seathru-dataset)

[UVEB](https://github.com/yzbouc/UVEB)

[DRUVA](https://github.com/nishavarghese15/DRUVA)

## Benchmark

[Benchm](https://github.com/CXH-Research/GuidedHybSensUIR)

## Method

### Traditional

| Title                                                        | Year | Publication                         | Code                                                         |

| ------------------------------------------------------------ | ---- | ----------------------------------- | ------------------------------------------------------------ |

| Enhancing the low quality images using Unsupervised Colour Correction Method | 2010 | SMC                                 | [Python](https://github.com/wangyanckxx/Single-Underwater-Image-Enhancement-and-Color-Restoration) |

| Underwater Image Enhancement by Wavelength Compensation and Dehazing | 2011 | TIP                                 | [MATLAB](https://github.com/mohitkumarahuja/Underwater-Image-Enhancement-by-Wavelength-Compensation-and-Dehazing) |

| Enhancing underwater images and videos by fusion             | 2012 | CVPR                                | [MATLAB](https://github.com/wangyanckxx/Single-Underwater-Image-Enhancement-and-Color-Restoration) |

| Transmission Estimation in Underwater Single Images          | 2013 | ICCV Workshop                       | [Python](https://github.com/wangyanckxx/Single-Underwater-Image-Enhancement-and-Color-Restoration) |

| Single underwater image enhancement using depth estimation based on blurriness | 2015 | ICIP                                | [MATLAB](https://github.com/ytpeng-aimlab/Single-Underwater-Image-Enhancement-Using-Depth-Estimation-based-on-Blurriness) |

| Automatic Red-Channel Underwater Image Restoration           | 2015 | JVCIR                               | [Python](https://github.com/agaldran/UnderWater)             |

| Underwater image enhancement by dehazing with minimum information loss and histogram distribution prior | 2016 | TIP                                 | [MATLAB](https://github.com/Li-Chongyi/TIP2016-code)         |

| Single underwater image restoration by blue-green channels dehazing and red channel correction | 2016 | ICASSP                              | [Python](https://github.com/wangyanckxx/Single-Underwater-Image-Enhancement-and-Color-Restoration) |

| Underwater Image Restoration Based on Image Blurriness and Light Absorption | 2017 | TIP                                 | [Python](https://github.com/wangyanckxx/Single-Underwater-Image-Enhancement-and-Color-Restoration) |

| Diving into Haze-Lines: Color Restoration of Underwater Images | 2017 | BMVC                                | [MATLAB](https://github.com/danaberman/underwater-hl)        |

| Shallow-Water Image Enhancement Using Relative Global Histogram Stretching Based on Adaptive Parameter Acquisition | 2018 | MMM                                 | [Python](https://github.com/Amruth-sagar/Shallow-water-Image-Enhancement) |

| Color Balance and Fusion for Underwater Image Enhancement    | 2018 | TIP                                 | [MATLAB](https://github.com/fergaletto/Color-Balance-and-fusion-for-underwater-image-enhancement.-.) |

| Enhancement of Underwater Images With Statistical Model of Background Light and Optimization of Transmission Map | 2019 | IEEE T BROADCAST                    | [Python](https://github.com/wangyanckxx/Enhancement-of-Underwater-Images-with-Statistical-Model-of-BL-and-Optimization-of-TM) |

| Real-time Model-based Image Color Correction for Underwater Robots | 2019 | IROS                                | [C++](https://github.com/dartmouthrobotics/underwater_color_enhance) |

| Sea-Thru: A Method for Removing Water From Underwater Images | 2019 | CVPR                                | [Python](https://github.com/hainh/sea-thru)                  |

| Restoration of Non-rigidly Distorted Underwater Images using a Combination of Compressive Sensing and Local Polynomial Image Representations | 2019 | ICCV                                | [MATLAB](https://github.com/jeringeo/CompressiveFlows)       |

| Local Color Mapping combined with Color Transfer for Underwater Image Enhancement | 2019 | WACV                                | [MATLAB](https://github.com/rprotasiuk/underwater_enhancement) |

| Adaptive Enhancement of Underwater Images using Multiobjective PSO | 2019 | MTA                                 | [MATLAB](https://github.com/Rajni27/CIEUI)                   |

| L^2UWE: A Framework for the Efficient Enhancement of Low-Light Underwater Images Using Local Contrast and Multi-Scale Fusion | 2020 | CVPR Workshop                       | [MATLAB](https://github.com/tunai/l2uwe)                     |

| A novel dark channel prior guided variational framework for underwater image restoration | 2020 | JVCIR                               | [MATLAB](https://github.com/Hou-Guojia/UTV)                  |

| Bayesian retinex underwater image enhancement                | 2021 | EAAI                                | [MATLAB](https://github.com/zhuangpeixian/Bayesian-Retinex-Underwater-Enhancement) |

| TEBCF: Real-World Underwater Image Texture Enhancement Model Based on Blurriness and Color Fusion | 2021 | TGRS                                | [MATLAB](https://github.com/bilityniu/TEBCF_tgrs)\|[Python](https://github.com/evan-person/TEBCF_python) |

| An Underwater Image Vision Enhancement Algorithm Based on Contour Bougie Morphology | 2021 | TGRS                                | [MATLAB](https://github.com/bilityniu/cbm_tgrs)              |

| Underwater Image Enhancement by Attenuated Color Channel Correction and Detail Preserved Contrast Enhancement | 2022 | IEEE Journal of Oceanic Engineering | [MATLAB](https://github.com/Li-Chongyi/JOE2021_ACDC)         |

| Underwater Image Enhancement via Minimal Color Loss and Locally Adaptive Contrast Enhancement | 2022 | TIP                                 | [MATLAB](https://github.com/Li-Chongyi/MMLE_code)            |

| Underwater Image Enhancement With Hyper-Laplacian Reflectance Priors | 2022 | TIP                                 | [MATLAB](https://github.com/zhuangpeixian/HLRP)              |

| A Perception-Aware Decomposition and Fusion Framework for Underwater Image Enhancement | 2022 | TCSVT                               | [MATLAB](https://github.com/59Kkk/SPDF)                      |

| A Variational Framework for Underwater Image Dehazing and Deblurring | 2022 | TCSVT                               | [MATLAB](https://github.com/Hou-Guojia/UNTV)                 |

| Underwater Image Enhancement via Piecewise Color Correction and Dual Prior Optimized Contrast Enhancement | 2022 | SPL                                 | [MATLAB](https://github.com/Li-Chongyi/PCDE)                 |

| Towards Underwater Image Restoration: A Physical-accurate Pipeline and a Large Scale Full-reference Benchmark | 2022 | ICME                                | [Python](https://github.com/TongJiayan/MonoUIR)              |

| Non-uniform Illumination Underwater Image Restoration via Illumination Channel Sparsity Prior | 2023 | TCSVT                               | [MATLAB](https://github.com/Hou-Guojia/ICSP)                 |

| Underwater Image Enhancement via Weighted Wavelet Visual Perception Fusion | 2023 | TCSVT                               | [MATLAB](https://github.com/Li-Chongyi/WWPF_code)            |

| Underwater camera: Improving visual perception via adaptive dark pixel prior and color correction | 2023 | IJCV                                | [Python](https://github.com/zhoujingchun03/ADPCC_Code)       |

| Rank-One Prior: Real-time Scene Recovery                     | 2023 | TPAMI                               | [MATLAB](https://github.com/junliumath/ROP)                  |

| Underwater Image Enhancement via Piecewise Color Correction and Dual Prior Optimized Contrast Enhancement | 2023 | SPL                                 | [MATLAB](https://github.com/Li-Chongyi/PCDE)                 |

| Enhancing Underwater Image via Adaptive Color and Contrast Enhancement, and Denoising | 2023 | EAAI                                | [MATLAB](https://github.com/Hou-Guojia/ACCE-D)               |

| HFM: A hybrid fusion method for underwater image enhancement | 2024 | EAAI                                | [MATLAB](https://github.com/An-Shunmin/HFM)                  |

| Underwater Color Disparities: Cues for Enhancing Underwater Images Toward Natural Color Consistencies | 2024 | TCSVT                               | [Python](https://gitee.com/wanghaoupc/Underwater_Color_Disparities) |

| Underwater Image Enhancement via Adaptive Color Correction and Stationary Wavelet Detail Enhancement | 2024 | Access                              | [MATLAB](https://github.com/Zhenbo-Wang/Adaptive-Color-Correction-and-Stationary-Wavelet-Detail-Enhancement) |

| CBLA: Color-Balanced Locally Adjustable Underwater Image Enhancement | 2024 | TIM                                 | [MATLAB](https://github.com/Severus-Matthew/CBLA)            |

| Polarization-Driven Solution for Mitigating Scattering and Uneven Illumination in Underwater Imagery | 2024 | TGRS                                | [MATLAB](https://github.com/ShenLH93/PDS)                    |

### Learning

| Name                           | Year | Publication                         | Code                                                         |

| ------------------------------ | ---- | ----------------------------------- | ------------------------------------------------------------ |

| WaterGAN                       | 2017 | RA-L                                | [Tensorflow](https://github.com/kskin/WaterGAN)              |

| UGAN                           | 2018 | ICRA                                | [Tensorflow](https://github.com/cameronfabbri/Underwater-Color-Correction)\|[Pytorch](https://github.com/xahidbuffon/FUnIE-GAN) |

| WAug Encoder-Decoder           | 2019 | CVPR Workshop                       | [Pytorch](https://github.com/AdarshMJ/Underwater-Image-Enhancement-via-Style-Transfer) |

| UIE-DAL                        | 2019 | CVPR Workshops                      | [Pytorch](https://github.com/VITA-Group/All-In-One-Underwater-Image-Enhancement-using-Domain-Adversarial-Learning) |

| WaterNet                       | 2019 | TIP                                 | [Tensorflow](https://github.com/Li-Chongyi/Water-Net_Code)\|[Pytorch](https://github.com/BIGWangYuDong/UWEnhancement) |

| GAN-RS                         | 2019 | IEEE T IND ELECTRON                 | [Pytorch](https://github.com/SeanChenxy/GAN_RS)              |

| UResnet                        | 2019 | Access                              | [Pytorch](https://github.com/OucMLLab-LiuPeng/UResnet)       |

| cGAN                           | 2019 | SPIC                                | [Tensorflow](https://github.com/JOU-UIP/Underwater-image-enhancement-based-on-conditional-generative-adversarial-network) |

| UW-Net                         | 2019 | ICIP                                | [Tensorflow](https://github.com/honeygupta/UW-Net)           |

| FUnIE-GAN                      | 2020 | RA-L                                | [Tensorflow & Pytorch](https://github.com/xahidbuffon/FUnIE-GAN) |

| SESR                           | 2020 | RSS                                 | [Tensorflow](https://github.com/xahidbuffon/Deep_SESR)       |

| UWCNN                          | 2020 | PR                                  | [Tensorflow](https://github.com/saeed-anwar/UWCNN)\|[Pytorch](https://github.com/BIGWangYuDong/UWEnhancement) |

| HybridDetectionGAN             | 2020 | TCSVT                               | [Tensorflow](https://github.com/cjh666-sym/HybridDetectionGAN) |

| cGAN                           | 2020 | SPIC                                | [Tensorflow](https://github.com/JOU-UIP/Underwater-image-enhancement-based-on-conditional-generative-adversarial-network) |

| PRWNet                         | 2021 | ICCV Workshop                       | [Pytorch](https://github.com/huofushuo/PRWNet)               |

| CWR                            | 2021 | IGARSS                              | [Pytorch](https://github.com/JunlinHan/CWR)                  |

| Shallow-UWNet                  | 2021 | AAAI                                | [Pytorch](https://github.com/mkartik/Shallow-UWnet)          |

| UColor                         | 2021 | TIP                                 | [Tensorflow](https://github.com/Li-Chongyi/Ucolor)\|[Pytorch](https://github.com/CV-Reimplementation/Ucolor-Reimplementation) |

| UIEC^2-Net                     | 2021 | SPIC                                | [Pytorch](https://github.com/BIGWangYuDong/UWEnhancement)    |

| PFATL                          | 2022 | IEEE Journal of Oceanic Engineering | [Pytorch](https://github.com/tjulab540/UIE)                  |

| STSC                           | 2022 | ICRA                                | [Pytorch](https://github.com/wdhudiekou/STSC)                |

| URSCT-SESR                     | 2022 | TGRS                                | [Pytorch](https://github.com/TingdiRen/URSCT-SESR)           |

| TACL                           | 2022 | TIP                                 | [Pytorch](https://github.com/Jzy2017/TACL)                   |

| TOPAL                          | 2022 | TCSVT                               | [Pytorch](https://github.com/Jzy2017/TOPAL)                  |

| SGUIE-Net                      | 2022 | TIP                                 | [Pytorch](https://github.com/trentqq/SGUIE-Net_Simple)       |

| PUIE-Net                       | 2022 | ECCV                                | [Pytorch](https://github.com/zhenqifu/puie-net)              |

| UIE-WD                         | 2022 | ICASSP                              | [Pytorch](https://github.com/ZZiyin/UIE-WD_Code)             |

| SCNet                          | 2022 | ICASSP                              | [Pytorch](https://github.com/zhenqifu/SCNet)                 |

| UHD-SFNet                      | 2022 | ACCV                                | [Pytorch](https://github.com/zzr-idam/UHD-Underwater-Image-Enhancement) |

| AutoEnhancer                   | 2022 | ACCV                                | [Pytorch](https://github.com/piggy2009/autoEnhancer)         |

| ADMNNet                        | 2022 | KBS                                 | [Pytorch](https://github.com/xhyan006/ADMNNet)               |

| LANet                          | 2022 | RA-L                                | [Pytorch](https://github.com/LiuShiBen/LANet)                |

| SIBM                           | 2022 | MM                                  | [Pytorch](https://github.com/IntegralCoCo/SIBM)              |

| MBANet                         | 2022 | PR                                  | [Pytorch](https://github.com/lizexuan1/MBANet)               |

| USUIR                          | 2022 | AAAI                                | [Pytorch](https://github.com/zhenqifu/USUIR)                 |

| UIR-Net                        | 2022 | Remote Sensing                      | [Pytorch](https://github.com/zhangbaijin/UIR-Net)            |

| UDAformer                      | 2022 | C & G                               | [Pytorch](https://github.com/ShenZhen0502/UDAformer)         |

| PHYSICAL IMAGE FORMATION MODEL | 2022 | ICASSP                              | [Pytorch](https://github.com/chaishu321/UNSUPERVISED-AND-UNTRAINED-UNDERWATER-IMAGE-RESTORATION-BASED-ON-PHYSICAL-IMAGE-FORMATION-MODEL) |

| UIESS                          | 2022 | Access                              | [Pytorch](https://github.com/fordevoted/UIESS)               |

| GSL                            | 2022 | TVC                                 | [Pytorch](https://github.com/CV-Reimplementation/Global-structure-guided-learning-framework-for-underwater-image-enhancement) |

| U-Shape Transformer            | 2023 | TIP                                 | [Pytorch](https://github.com/LintaoPeng/U-shape_Transformer_for_Underwater_Image_Enhancement) |

| PUGAN                          | 2023 | TIP                                 | [Pytorch](https://github.com/rmcong/PUGAN_TIP2023)           |

| Semi-UIR                       | 2023 | CVPR                                | [Pytorch](https://github.com/Huang-ShiRui/Semi-UIR)          |

| SyreaNet                       | 2023 | ICRA                                | [Pytorch](https://github.com/RockWenJJ/SyreaNet)             |

| Five A+ Network                | 2023 | BMVC                                | [Pytorch](https://github.com/Owen718/FiveAPlus-Network)      |

| UIALN                          | 2023 | TCSVT                               | [Pytorch](https://github.com/kevendai/UIALN-Reproduction-Experiment) |

| Deep WaveNet                   | 2023 | TOMM                                | [Pytorch](https://github.com/pksvision/Deep-WaveNet-Underwater-Image-Restoration) |

| DM                             | 2023 | MM                                  | [Pytorch](https://github.com/piggy2009/DM_underwater)        |

| P2CNet                         | 2023 | TCSVT                               | [Pytorch](https://github.com/Ray2OUC/P2CNet)                 |

| TCTL-Net                       | 2023 | TCSVT                               | [Pytorch](https://github.com/trentqq/TCTL-Net)               |

| CLUIE-Net                      | 2023 | TCSVT                               | [Pytorch](https://github.com/justwj/CLUIE-Net)               |

| HAAM-GAN                       | 2023 | EAAI                                | [Pytorch](https://github.com/zhoujingchun03/HAAM-GAN)        |

| ASNet+ANNet                    | 2023 | EAAI                                | [Pytorch](https://github.com/cwoop92/Underwater-image-enhancement-using-adaptive-standardization-and-normalization-networks) |

| LiteEnhanceNet                 | 2023 | ESWA                                | [Pytorch](https://github.com/zhangsong1213/LiteEnhanceNet)   |

| GUPDM                          | 2023 | MM                                  | [Pytorch](https://github.com/shiningZZ/GUPDM)                |

| Boths                          | 2023 | GRSL                                | [Pytorch](https://github.com/perseveranceLX/BOths)           |

| TUDA                           | 2023 | TIP                                 | [Pytorch](https://github.com/zyWang-Power/TUDA)              |

| Spectroformer                  | 2023 | WACV                                | [Pytorch](https://github.com/Mdraqibkhan/Spectroformer)      |

| RAUNE-Net                      | 2023 | IFTC                                | [Pytorch](https://github.com/fansuregrin/RAUNE-Net)          |

| Joint_ID                       | 2023 | IEEE Sensors                        | [Pytorch](https://github.com/sparolab/Joint_ID)              |

| UIEPTA                         | 2023 | IJCNN                               | [Pytorch](https://github.com/Mdraqibkhan/UIEPTA)             |

| NU2Net                         | 2023 | AAAI                                | [Pytorch](https://github.com/RQ-Wu/UnderwaterRanker)         |

| DAUT                           | 2023 | ICIP                                | [Pytorch](https://github.com/MBadran2000/Depth-Aware-U-shape-Transformer) |

| ACU-Net                        | 2024 | SIVP                                | [Pytorch](https://github.com/666ShuqiWang/Underwater-Image-Enhancement-by-Combining-Multi-Attention-with-Recurrent-Residual-Convolutional-UNet) |

| UWFormer                       | 2024 | IJCNN                               | [Pytorch](https://github.com/leiyingtie/UWFormer)            |

| DICAM                          | 2024 | Neurocomputing                      | [Pytorch](https://github.com/hfarhaditolie/DICAM)            |

| INSPIRATION                    | 2024 | EAAI                                | [Pytorch](https://gitee.com/wanghaoupc/uie_inspiration)      |

| AoSRNet                        | 2024 | KBS                                 | [Pytorch](https://github.com/LouisYuxuLu/AoSRNet)            |

| IACC                           | 2024 | TGRS                                | [Pytorch](https://github.com/zhoujingchun03/IACC)            |

| TFUIE                          | 2024 | TGRS                                | [Pytorch](https://github.com/gemyumeng/TFUIE)                |

| DiffWater                      | 2024 | JSTAR                               | [Pytorch](https://github.com/Guan-MS/DiffWater)              |

| SFGNet                         | 2024 | ICASSP                              | [Pytorch](https://github.com/zhihefang/SFGNet)               |

| WsUIDNet                       | 2024 | TGRS                                | [Pytorch](https://github.com/Wangxy97/WsUID-Net_code)        |

| CCMSRNet                       | 2024 | TGRS                                | [Pytorch](https://github.com/INDTLab/CCMSRNet)               |

| SimUESR                        | 2024 | Neurocomputing                      | [Pytorch](https://github.com/lpm1001/SimUESR)                |

| MDP                            | 2024 | IEEE Journal of Oceanic Engineering | [Tensorflow](https://gitee.com/sunshixin_upc/underwater-image-enhancement-with-reinforcement-learning) |

| CECF                           | 2024 | AAAI                                | [Pytorch](https://github.com/Xiaofeng-life/CECF)             |

| DGD-cGAN                       | 2024 | PR                                  | [Pytorch](https://github.com/SalPGS/DGD-cGAN)                |

| HCLR-Net                       | 2024 | IJCV                                | [Pytorch](https://github.com/zhoujingchun03/HCLR-Net)        |

| WPFNet                         | 2024 | KBS                                 | [Pytorch](https://github.com/LiuShiBen/WPFNet)               |

| GC-UIE                         | 2024 | KBS                                 | [Pytorch](https://github.com/xzx11/GC-UIE)                   |

| DDformer                       | 2024 | KBS                                 | [Pytorch](https://github.com/ZhiGao-hfuu/DDformer)           |

| DGD-cGAN                       | 2024 | PR                                  | [Pytorch](https://github.com/SalPGS/DGD-cGAN)                |

| UVE-Net                        | 2024 | CVPR                                | [Pytorch](https://github.com/yzbouc/UVEB)                    |

| UIE-UnFold                     | 2024 | DSAA                                | [Pytorch](https://github.com/CXH-Research/UIE-UnFold)        |

| CTM                            | 2024 | Applied Soft Computing              | [Pytorch](https://github.com/WindySprint/CTM)                |

| DIMN                           | 2024 | Scientific Reports                  | [Pytorch](https://github.com/wwaannggllii/DIMN)              |

| USIS-SAM                       | 2024 | ICML                                | [Pytorch](https://github.com/LiamLian0727/USIS10K)           |

| DedustGAN                      | 2024 | ESWA                                | [Pytorch](https://github.com/MXL696/DedustGAN)               |

| Osmosis                        | 2024 | ECCV                                | [Pytorch](https://github.com/osmosis-diffusion/osmosis-diffusion-code) |

| UIR-PolyKernel                 | 2024 | ICASSP                              | [Pytorch](https://github.com/CXH-Research/UIR-PolyKernel)    |

| CE-VAE                         | 2024 | WACV                                | [Pytorch](https://github.com/iN1k1/ce-vae-underwater-image-enhancement) |

| PixelMamba                     | 2024 | ACCV                                | [Pytorch](https://github.com/weitunglin/pixmamba)            |

| GuidedHybSensUIR               | 2025 | TCSVT                               | [Pytorch](https://github.com/CXH-Research/GuidedHybSensUIR)  |

| HUPE                           | 2025 | IJCV                                | [Pytorch](https://github.com/ZengxiZhang/HUPE)               |

| Phaseformer                    | 2025 | WACV                                | [Pytorch](https://github.com/Mdraqibkhan/Phaseformer)        |

| Udnet                          | 2025 | EAAI                                | [Pytorch](https://github.com/alzayats/UDnet)                 |

## Metric

### Full-Reference

[torchmetrics](https://github.com/Lightning-AI/torchmetrics) for cuda calculation

**PSNR**

**SSIM**

**MAE**

**MSE**

**RMSE**

### Non-Reference

[Metrics](https://github.com/rahim-khan-iitg/underwater_image_enhancement)

**UIQM**

**UCIQE**

**NIQE**

**PS**

**MUSIQ**
ecosyste.ms

Data

Tools

Indexes

Applications

Experiments

Awesome

https://github.com/CXH-Research/Underwater-Image-Enhancement

Awesome Lists containing this project

README
