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)
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.
Please first specify TRAIN_DIR, VAL_DIR and SAVE_DIR in section TESTING in config.yml.
python infer.py