This repository is the official PyTorch implementation of SGUIE-Net.
You need to prepare datasets for following training and testing activities, the detailed information is at Dataset Setup.
python train.py --dataroot /path_to_data --name train_name --model SGUIENet --display_env display_env_name
python test.py --dataroot /path_to_data --name test_name --model test_SGUIE
You can download the trained model from here.
@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}
}