Skip to content

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

Notifications You must be signed in to change notification settings

zhangbaijin/UIR-Net

 
 

Repository files navigation

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 image

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 image

To reproduce PSNR/SSIM scores of the paper, run

evaluate_PSNR_SSIM.m 

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages

  • Python 92.0%
  • MATLAB 8.0%