justwj / CLUIE-Net

Beyond Single Reference for Training: Underwater Image Enhancement via Comparative Learning

This repository includes two branches. This branch is the official PyTorch implementation of CLUIE-Net, another branch is the official PyTorch implementation of RQSD-Net.

Test_Demo

You should download the pretrained CLUIE-Net model from the following link, then put them in the folder './ckpt'.

Google Drive Link

Baidu Netdisk Link

python test.py --fe_load_path /path_to_ckpt --fI_load_path /path_to_ckpt 

You can find the enhanced results in folder './output'

Bibtex

@ARTICLE{2023Beyond,
  author={Li, Kunqian and Wu, Li and Qi, Qi and Liu, Wenjie and Gao, Xiang and Zhou, Liqin and Song, Dalei},
  journal={IEEE Transactions on Circuits and Systems for Video Technology}, 
  title={Beyond Single Reference for Training: Underwater Image Enhancement via Comparative Learning}, 
  year={2023},
  volume={33},
  number={6},
  pages={2561-2576},
  doi={10.1109/TCSVT.2022.3225376}}