Awesome
Awesome Underwater Image Enhancement
This is a resouce list for underwater image enhancement
Network Structure
| Model | Paper | Venue | Link |
|---|---|---|---|
| UWCNN | Underwater scene prior inspired deep underwater image and video enhancement | Pattern Recognition 2020 | PDF/Code |
| UWNet | Shallow-uwnet: Compressed model for underwater image enhancement (student abstract) | AAAI Conference on Artificial Intelligence 2021 | PDF/Code |
| UResNet | Underwater image enhancement with a deep residual framework | IEEE Access 2019 | PDF/Code |
| DUIR | Deep underwater image restoration and beyond | IEEE Signal Processing Letters 2020 | |
| FloodNet | FloodNet: Underwater image restoration based on residual dense learning | Signal Processing: Image Communication 2022 | |
| LAFFNet | Laffnet: A lightweight adaptive feature fusion network for underwater image enhancement | IEEE International Conference on Robotics and Automation 2021 | |
| UICoE-Net | Underwater image co-enhancement with correlation feature matching and joint learning | IEEE Transactions on Circuits and Systems for Video Technology 2021 | PDF/Code |
| PUIE-Net | Uncertainty inspired underwater image enhancement | European Conference on Computer Vision 2022 | PDF/Code |
| WaveNet | Wavelength-based attributed deep neural network for underwater image restoration | ACM Transactions on Multimedia Computing, Communications and Applications 2023 | PDF/Code |
| ADMNNet | Attention-guided dynamic multi-branch neural network for underwater image enhancement | Knowledge-Based Systems 2022 | PDF/Code |
| LightEnhanceNet | LiteEnhanceNet: A lightweight network for real-time single underwater image enhancement | Expert Systems with Applications 2024 | PDF/Code |
| HAAM-GAN | Hierarchical attention aggregation with multi-resolution feature learning for GAN-based underwater image enhancement | Engineering Applications of Artificial Intelligence 2023 | PDF/Code |
| MFEF | Multi-view underwater image enhancement method via embedded fusion mechanism | Engineering Applications of Artificial Intelligence 2023 | |
| PTT | Underwater image enhancement using pre-trained transformer | International Conference on Image Analysis and Processing 2022 | |
| U-Transformer | U-shape transformer for underwater image enhancement | IEEE Transactions on Image Processing 2023 | PDF/Code |
| UWAGA | Underwater image enhancement via adaptive group attention-based multiscale cascade transformer | IEEE Transactions on Instrumentation and Measurement 2022 | |
| WaterFormer | WaterFormer: A Global--Local Transformer for Underwater Image Enhancement With Environment Adaptor | IEEE Robotics & Automation Magazine 2024 | |
| Spectroformer | Spectroformer: Multi-Domain Query Cascaded Transformer Network for Underwater Image Enhancement | IEEE Winter Conference on Applications of Computer Vision 2024 | |
| UHD | UHD Underwater Image Enhancement via Frequency-Spatial Domain Aware Network | Proceedings of the Asian Conference on Computer Vision 2022 | PDF/Code |
| WFI2-Net | Wavelet-based Fourier Information Interaction with Frequency Diffusion Adjustment for Underwater Image Restoration | arXiv preprint arXiv:2311.16845 2023 | PDF/Code |
| TANet | TANet: Transmission and atmospheric light driven enhancement of underwater images | Expert Systems with Applications 2024 | |
| UIE-INN | Underwater Image Enhancement and Super-Resolution Using Implicit Neural Networks | IEEE International Conference on Image Processing 2023 | |
| SFGNet | Toward Sufficient Spatial-Frequency Interaction for Gradient-aware Underwater Image Enhancement | IEEE International Conference on Acoustics, Speech and Signal Processing 2024 | PDF/Code |
| UIE-WD | A wavelet-based dual-stream network for underwater image enhancement | IEEE International Conference on Acoustics, Speech and Signal Processing 2022 | PDF/Code |
| EUIE | Exemplar-based underwater image enhancement augmented by wavelet corrected transforms | IEEE Conference on Computer Vision and Pattern Recognition Workshops 2019 | |
| MWEN | A multi-level wavelet-based underwater image enhancement network with color compensation prior | Expert Systems with Applications 2024 | |
| AutoEnhancer | AutoEnhancer: Transformer on U-Net Architecture Search for Underwater Image Enhancement | Proceedings of the Asian Conference on Computer Vision 2022 | PDF/Code |
Learning Strategy
| Model | Paper | Venue | Link |
|---|---|---|---|
| DGD-cGAN | DGD-cGAN: A dual generator for image dewatering and restoration | Pattern Recognition 2024 | PDF/Code |
| FGAN | A fusion adversarial underwater image enhancement network with a public test dataset | arXiv preprint arXiv:1906.06819 2019 | |
| UIE-cGAN | Underwater image enhancement based on conditional generative adversarial network | Signal Processing: Image Communication 2020 | |
| TOPAL | Target oriented perceptual adversarial fusion network for underwater image enhancement | IEEE Transactions on Circuits and Systems for Video Technology 2022 | PDF/Code |
| EUIGAN | Enhancing underwater imagery using generative adversarial networks | International Conference on Robotics and Automation 2018 | |
| CE-CGAN | Contrast enhancement of underwater images using conditional generative adversarial network | Multimedia Tools and Applications 2023 | PDF/Code |
| RUIG | Ruig: Realistic underwater image generation towards restoration | IEEE Conference on Computer Vision and Pattern Recognition Workshops 2021 | |
| URanker | Underwater ranker: Learn which is better and how to be better | AAAI Conference on Artificial Intelligence 2023 | PDF/Code |
| PDD-Net | Perception-Driven Deep Underwater Image Enhancement without Paired Supervision | IEEE Transactions on Multimedia 2023 | |
| CLUIE-Net | Beyond single reference for training: underwater image enhancement via comparative learning | IEEE Transactions on Circuits and Systems for Video Technology 2022 | |
| TACL | Twin adversarial contrastive learning for underwater image enhancement and beyond | IEEE Transactions on Image Processing 2022 | PDF/Code |
| Semi-UIR | Contrastive semi-supervised learning for underwater image restoration via reliable bank | IEEE Conference on Computer Vision and Pattern Recognition 2023 | PDF/Code |
| CWR | Single underwater image restoration by contrastive learning | IEEE International Geoscience and Remote Sensing Symposium 2021 | PDF/Code |
| DRDCL | Unsupervised Underwater Image Enhancement Based on Disentangled Representations via Double-Order Contrastive Loss | IEEE Transactions on Geoscience and Remote Sensing 2024 | |
| TFUIE | Task-Friendly Underwater Image Enhancement for Machine Vision Applications | IEEE Transactions on Geoscience and Remote Sensing 2023 | |
| CL-UIE | Deformable convolutional network constrained by contrastive learning for underwater image enhancement | IEEE Geoscience and Remote Sensing Letters 2023 | |
| HCLR-Net | HCLR-Net: Hybrid Contrastive Learning Regularization with Locally Randomized Perturbation for Underwater Image Enhancement | International Journal of Computer Vision 2024 | PDF/Code |
| UIE-CWR | Underwater image restoration via contrastive learning and a real-world dataset | Remote Sensing 2022 | PDF/Code |
| HPUIE-RL | A hierarchical probabilistic underwater image enhancement model with reinforcement tuning | Journal of Visual Communication and Image Representation 2024 |
Learning Stage
| Model | Paper | Venue | Link |
|---|---|---|---|
| MBANet | Investigating intrinsic degradation factors by multi-branch aggregation for real-world underwater image enhancement | Pattern Recognition 2023 | |
| GSL | Global structure-guided learning framework for underwater image enhancement | The Visual Computer 2022 | PDF/Code |
| CUIE | Coarse-to-fine Underwater Image Enhancement with Lightweight CNN and Attention-based Refinement | Journal of Visual Communication and Image Representation 2024 | |
| DMML | DMML: Deep Multi-Prior and Multi-Discriminator Learning for Underwater Image Enhancement | IEEE Transactions on Broadcasting 2024 | |
| UIE-DM | Underwater image enhancement by transformer-based diffusion model with non-uniform sampling for skip strategy | ACM International Conference on Multimedia 2023 | PDF/Code |
| SU-DDPM | Speed-Up DDPM for real-time underwater image enhancement | IEEE Transactions on Circuits and Systems for Video Technology 2023 | PDF/Code |
| CPDM | CPDM: Content-Preserving Diffusion Model for Underwater Image Enhancement | arXiv preprint arXiv:2401.15649 2024 | PDF/ |
Assistance Task
| Model | Paper | Venue | Link |
|---|---|---|---|
| SGUIE-Net | SGUIE-Net: Semantic attention guided underwater image enhancement with multi-scale perception | IEEE Transactions on Image Processing 2022 | PDF/Code |
| SATS | Semantic-aware Texture-Structure Feature Collaboration for Underwater Image Enhancement | International Conference on Robotics and Automation 2022 | PDF/Code |
| DAL-UIE | Domain Adversarial Learning Towards Underwater Image Enhancement | IEEE International Conference on Computer Vision Workshops 2023 | |
| WaterFlow | Waterflow: heuristic normalizing flow for underwater image enhancement and beyond | ACM International Conference on Multimedia 2023 | |
| HDGAN | Perceptual underwater image enhancement with deep learning and physical priors | IEEE Transactions on Circuits and Systems for Video Technology 2020 | |
| Joint-ID | Joint-ID: Transformer-based Joint Image Enhancement and Depth Estimation for Underwater Environments | IEEE Sensors Journal 2023 | PDF/Code |
| DAUT | DAUT: Underwater Image Enhancement Using Depth Aware U-shape Transformer | IEEE International Conference on Image Processing 2023 | PDF/Code |
| DepthCue | DepthCue: Restoration of Underwater Images Using Monocular Depth as a Clue | IEEE Winter Conference on Applications of Computer Vision Workshops 2023 | |
| HybrUR | Hybrur: A hybrid physical-neural solution for unsupervised underwater image restoration | IEEE Transactions on Image Processing 2023 | PDF/Code |
Domain Perspective
| Model | Paper | Venue | Link |
|---|---|---|---|
| TUDA | Domain adaptation for underwater image enhancement | IEEE Transactions on Image Processing 2023 | |
| TSDA | Two-step domain adaptation for underwater image enhancement | Pattern Recognition 2022 | |
| DAAL | Domain adaptive adversarial learning based on physics model feedback for underwater image enhancement | arXiv preprint arXiv:2002.09315 2020 | |
| IUIE | Domain Adaptation for In-Air to Underwater Image Enhancement via Deep Learning | IEEE Transactions on Emerging Topics in Computational Intelligence 2023 | |
| TRUDGCR | Towards Realistic Underwater Dataset Generation and Color Restoration | Proceedings of the Thirteenth Indian Conference on Computer Vision, Graphics and Image Processing 2022 | |
| UW-CycleGAN | UW-CycleGAN: Model-driven CycleGAN for Underwater Image Restoration | IEEE Transactions on Geoscience and Remote Sensing 2023 | |
| URD-UIE | Unsupervised underwater image enhancement via content-style representation disentanglement | Engineering Applications of Artificial Intelligence 2023 | |
| TACL | Twin adversarial contrastive learning for underwater image enhancement and beyond | IEEE Transactions on Image Processing 2022 | PDF/Code |
| UIESS | Domain Adaptation for Underwater Image Enhancement via Content and Style Separation | IEEE Access 2022 | PDF/Code |
| UMRD | Unsupervised Multiple Representation Disentanglement Framework for Improved Underwater Visual Perception | IEEE Journal of Oceanic Engineering 2023 | |
| PWAE | Pixel-wise wasserstein autoencoder for highly generative dehazing | IEEE Transactions on Image Processing 2021 |
Disentanglement & Fusion
| Model | Paper | Venue | Link |
|---|---|---|---|
| IPMGAN | IPMGAN: Integrating physical model and generative adversarial network for underwater image enhancement | Neurocomputing 2021 | |
| ACPAB | Attenuation coefficient guided two-stage network for underwater image restoration | IEEE Signal Processing Letters 2020 | |
| USUIR | Unsupervised underwater image restoration: From a homology perspective | AAAI Conference on Artificial Intelligence 2022 | PDF/Code |
| AquaGAN | Aquagan: Restoration of underwater images | IEEE Conference on Computer Vision and Pattern Recognition Workshops 2022 | |
| PhysicalNN | Underwater image enhancement based on deep learning and image formation model | arXiv preprint arXiv:2101.00991 2021 | PDF/Code |
| GUPDM | A generalized physical-knowledge-guided dynamic model for underwater image enhancement | ACM International Conference on Multimedia 2023 | PDF/Code |
| CCMSR-Net | Deep Color-Corrected Multi-scale Retinex Network for Underwater Image Enhancement | IEEE Transactions on Geoscience and Remote Sensing 2023 | |
| ReX-Net | ReX-Net: A reflectance-guided underwater image enhancement network for extreme scenarios | Expert Systems with Applications 2023 | |
| ASSU-Net | Underwater Image Enhancement with an Adaptive Self Supervised Network | IEEE International Conference on Multimedia and Expo 2023 | |
| UGIF-Net | Ugif-net: An efficient fully guided information flow network for underwater image enhancement | IEEE Transactions on Geoscience and Remote Sensing 2023 | |
| TCTL-Net | TCTL-Net: Template-free Color Transfer Learning for Self-Attention Driven Underwater Image Enhancement | IEEE Transactions on Circuits and Systems for Video Technology 2023 | PDF/Code |
| MTNet | MTNet: A multi-task cascaded network for underwater image enhancement | Multimedia Tools and Applications 2023 | |
| UDNet | Adaptive uncertainty distribution in deep learning for unsupervised underwater image enhancement | arXiv preprint arXiv:2212.08983 2022 | PDF/Code |
| JLCL-Net | Joint luminance and chrominance learning for underwater image enhancement | IEEE Signal Processing Letters 2021 | |
| TBDNN | Two-branch deep neural network for underwater image enhancement in HSV color space | IEEE Signal Processing Letters 2021 | |
| MSTAF | A real-world underwater image enhancement method based on multi-color space and two-stage adaptive fusion | Signal, Image and Video Processing 2024 | |
| UColor | Underwater image enhancement via medium transmission-guided multi-color space embedding | IEEE Transactions on Image Processing 2021 | PDF/Code |
| SCNet | Underwater image enhancement via learning water type desensitized representations | IEEE International Conference on Acoustics, Speech and Signal Processing 2022 | PDF/Code |
| DAL | All-in-One Underwater Image Enhancement Using Domain-Adversarial Learning | IEEE Conference on Computer Vision and Pattern Recognition Workshops 2019 | PDF/Code |
| IACC | IACC: Cross-Illumination Awareness and Color Correction for Underwater Images Under Mixed Natural and Artificial Lighting | IEEE Transactions on Geoscience and Remote Sensing 2024 | PDF/Code |
| WaterNet | An underwater image enhancement benchmark dataset and beyond | IEEE Transactions on Image Processing 2019 | PDF/Code |
| MFEF | Multi-view underwater image enhancement method via embedded fusion mechanism | Engineering Applications of Artificial Intelligence 2023 | |
| F2UIE | F2UIE: feature transfer-based underwater image enhancement using multi-stackcnn | Multimedia Tools and Applications 2023 |