Home

Awesome

SGUIE-Net: Semantic Attention Guided Underwater Image Enhancement with Multi-Scale Perception

This repository is the official PyTorch implementation of SGUIE-Net.

Dataset preparation

You need to prepare datasets for following training and testing activities, the detailed information is at Dataset Setup.

Train

python train.py --dataroot /path_to_data --name train_name --model SGUIENet --display_env display_env_name

Test

python test.py --dataroot /path_to_data --name test_name --model test_SGUIE 

You can download the trained model from here.

Citation

@article{qi2022sguie,
  title={SGUIE-Net: Semantic Attention Guided Underwater Image Enhancement with Multi-Scale Perception},
  author={Qi, Qi and Li, Kunqian and Zheng, Haiyong and Gao, Xiang and Hou, Guojia and Sun, Kun},
  journal={IEEE Transactions on Image Processing},
  volume={31},
  pages={6816--6830},
  year={2022},
  publisher={IEEE}
}

Acknowledgements