Results of LUNet on underwater image restoration and enhancement

Training

• Download the UIEBD-with-snow

• Train the model with default arguments by running

python train_plus.py

Evaluation

1. Download the model and place it in ./pretrained_models/

2. Run

python test_plus.py

The enhancment and restoration results is

To reproduce PSNR/SSIM scores of the paper, run

evaluate_PSNR_SSIM.m