Awesome_Underwater_Image_Enhancement-UIE-

A repository of the latest work related to underwater image enhancement (awaiting continuous updates). We hope this can be of assistance!

[In Construction] Questions and discussions are most welcome! Upcoming works will be updated on a regular basis, feel free to contact me to add... 👍

Underwater Image Datasets

Image Enhancement / Restoration

• LSUI dataset: Data, Paper. (paired, Utrans)
• SUIM-E dataset: Data, Paper. (paired, SGUIE-Net)
• UFO dataset: Data, Paper. (paired, super-resolution, SESR)
• EUVP dataset: Data, Paper, Code. (paired and unpaired data; FUnIE-GAN)
• UIEB / UIEBD dataset: Data, Paper, Code. (Water-Net)
• UVEB dataset: Data, Paper, Code. (video data; UVE-Net)
• SQUID dataset: Data, Paper, Code. (Underwater-HL)
• U-45: Data, Paper. (UDAE)
• RUIE benchmark: Data, Paper. (RUIE-Net)
• Jamaica port royal: Data, Paper, Code. (Water-GAN)
• Virtual periscope: Data, Paper.
• Color restoration: Data, Paper, Code.
• TURBID data: Data, Paper.
• OceanDark dataset: Data, Paper.
• Underwater imagenet: Data, Paper, Code. (paired data; UGAN)

Others (super-resolution and ...)

Pelase go to https://github.com/xahidbuffon/Awesome_Underwater_Datasets which also includes image enhancement, super-resoulution, object detection, image segmentation and other datasets.

Papers and Codes

Underwater Image Enhancement

• CVPR2024 UVEB: A Large-scale Benchmark and Baseline Towards Real-World Underwater Video Enhancement [Paper] [Code]

• BMVC2023 Five A+ Network: You Only Need 9KParameters for Underwater Image Enhancement [Paper] [Code]

• TIP2023 PUGAN: Physical Model-Guided Underwater Image Enhancement Using GAN with Dual-Discriminators [Paper] [Code]

• TIP2022 U-shape Transformer for Underwater Image Enhancement [Paper] [Code]

• TIP2022 SGUIE-Net: Semantic Attention Guided Underwater Image Enhancement With Multi-Scale Perception [Paper] [Code]

• TGRS2022 Reinforced Swin-Convs Transformer for Simultaneous Underwater Sensing Scene Image Enhancement and Super-resolution [Paper] [Code]

• TIP2019 An Underwater Image Enhancement Benchmark Dataset and Beyond [Paper] [Code]

• TIP2020 Underwater Image Enhancement via Medium Transmission-Guided Multi-Color Space Embedding [Paper] [Code]

• RSS2020 Simultaneous Enhancement and Super-Resolution of Underwater Imagery for Improved Visual Perception [Paper] [Code]

Metrics

Full-Reference Evaluation

• PSNR (Peak Signal-to-Noise Ratio) [Wiki] [Matlab Code] [Python Code]
• SSIM (Structural similarity) [Wiki] [Matlab Code] [Python Code]

Non-Reference Evaluation

• UCIQE (Underwater Colour Image Quality Evaluation) A metric uses a linear combination of chromaticity, saturation, and contrast to quantify the uneven color cast, blurriness, and low contrast characteristics of underwater engineering and surveillance images. [Paper: An Underwater Colour Image Quality Evaluation Metric]
• UIQM (Underwater Image Quality Measure) includes three underwater image quality assessments: Underwater Image Colorfulness Measure (UICM), Underwater Image Sharpness Measure (UISM), and Underwater Image Contrast Measure (UIConM). [Paper: Human-Visual-System-Inspired Underwater Image Quality Measures]