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... 👍
- 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)
- 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)
Pelase go to https://github.com/xahidbuffon/Awesome_Underwater_Datasets which also includes image enhancement, super-resoulution, object detection, image segmentation and other datasets.
-
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]
- PSNR (Peak Signal-to-Noise Ratio) [Wiki] [Matlab Code] [Python Code]
- SSIM (Structural similarity) [Wiki] [Matlab Code] [Python Code]