zhangbaijin / UIR-Net

UIR-Net: A Simple and Effective Baseline for Underwater Image Restoration and Enhancement（Remote Sensing2022）

UIR-Net: A Simple and Effective Baseline for Underwater Image Restoration and Enhancement（Accepted in Remote Sensing2022）

Results of UIR-Net 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