GitHub - 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

python train_plus.py

python test_plus.py

The enhancment and restoration results is

evaluate_PSNR_SSIM.m

Name		Name	Last commit message	Last commit date
Latest commit History 34 Commits
checkpoint		checkpoint
model		model
utils		utils
CAM.py		CAM.py
DGUNet_plus.py		DGUNet_plus.py
README.md		README.md
SKmodel.py		SKmodel.py
config.py		config.py
data_RGB.py		data_RGB.py
dataset_RGB.py		dataset_RGB.py
evaluate_PSNR_SSIM.m		evaluate_PSNR_SSIM.m
introduction1.png		introduction1.png
losses.py		losses.py
results-UIEBD.png		results-UIEBD.png
test.py		test.py
test_plus.py		test_plus.py
train.py		train.py
train_plus.py		train_plus.py
training.yml		training.yml