The official repository for Wavelet-pixel domain progressive fusion network for underwater image enhancement.
Our work proposes an “Wavelet-pixel domain progressive fusion network for underwater image enhancement” (WPFNet). The proposed WPFNet progressively merge frequency features capturing fine-grained details and spatial features with rich color and illumination information.
- Python 3.8
- Pytorch 1.7.0
data
├── UIEB
│ └──trainA
│ └──trainB
├── Underwater-Dark
│ └──trainA
│ └──trainB
python Underwater_train.py --data_dir=./your dataset path
If you have any questions, please contact Shiben Liu at [liushiben@sia.cn].