Data & Code

Dataset

LSUI

LSUI is a large-scale underwater image dataset including 4279 image pairs, which covers more abundant underwater scenes(lighting conditions, water types and target categories) and better visual quality reference images than existing underwater datasets . The dataset contains 4279 real-world underwater images, and the corresponding clear images are generated as comparison references. We also provide the semantic segmentation map and medium transmission map for each image. You can download it from here(lsui). Our paper is under review, and once accepted, we will publish the corresponding semantic segmentation map and media transfer map.

​ Fig. 1: LSUI Dataset

Code

U-shape Transformer

You can find the official PyTorch implementation of U-shape Transformer for Underwater Image Enhancement. (arxiv, Dataset, video demo, visual results) in here. U-shape Transformer achieves state-of-the-art performance in underwater image enhancement task. Fig. 2: U-shape Transformer