UIR-PolyKernel

Underwater Image Restoration via Polymorphic Large Kernel CNNs

Xiaojiao Guo ^{👨‍💻‍} , Yihang Dong ^{👨‍💻‍} , Xuhang Chen , Weiwen Chen , Zimeng Li ^📮 , FuChen Zheng , Chi-Man Pun ^📮 ( 👨‍💻‍ Equal contributions, 📮 Corresponding author)

University of Macau, SIAT CAS, Huizhou Univeristy, Shenzhen Polytechnic University, The Hong Kong University of Science and Technology (Guangzhou), Baoshan Univeristy

In IEEE International Conference on Acoustics, Speech, and Signal Processing 2025 (ICASSP 2025)

⚙️ Usage

Training

You may download the dataset first, and then specify TRAIN_DIR, VAL_DIR and SAVE_DIR in the section TRAINING in config.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.

Inference

Please first specify TRAIN_DIR, VAL_DIR and SAVE_DIR in section TESTING in config.yml.

python infer.py

Citation