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)
git clone https://github.com/leiyingtie/UWFormer.git
cd UWFormer
pip install -r requirements.txt
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.
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.
If you find our work helpful for your research, please cite: