📖 UWFormer

UWFormer: Underwater Image Enhancement via a Semi-Supervised Multi-Scale Transformer

Weiwen Chen, Yingtie Lei, Shenghong Luo, Ziyang Zhou, Mingxian Li and Chi-Man Pun

University of Macau, Huizhou University

In International Joint Conference on Neural Networks 2024 (IJCNN 2024)

🔍 Usage

Installation

git clone https://github.com/leiyingtie/UWFormer.git
cd UWFormer
pip install -r requirements.txt

Training

Specify TRAIN_DIR, VAL_DIR and SAVE_DIR in the section TRAINING inconfig.yml.

For single GPU training:

python train.py

For multiple GPUs training:

accelerate config
accelerate launch train.py

If you have difficulties with the usage of accelerate, please refer to Accelerate.

💗 Acknowledgements

This work was supported in part by the Science and Technology Development Fund, Macau SAR, under Grant 0087/2020/A2 and Grant 0141/2023/RIA2.

💡 Citation

If you find our work helpful for your research, please cite: