Underwater-Image-Restoration

Two implementations are availabel in the repo:

#Instructions to run notebooks for Underwater-Image-Restoration

Fast Underwater Image Enhancement:

colab notebook: UIRF.ipynb

pretrained models for fuie (Fast underwater image enhancement) is saved in models

test data is available in: data/A output will be generated in data/output

colab notebook: UIRT.ipynb saved_models: pretrained models

test data is available in: test/input

output will be generated in test/output

Name		Name	Last commit message	Last commit date
Latest commit History 20 Commits
configs		configs
data/test/A		data/test/A
figs		figs
loss		loss
models		models
net		net
nets		nets
pytorch-ssim-loss		pytorch-ssim-loss
pytorch_ssim		pytorch_ssim
test		test
utility		utility
utils		utils
README.md		README.md
UIRF.ipynb		UIRF.ipynb
UIRT.ipynb		UIRT.ipynb
requirements.txt		requirements.txt
train_funiegan.py		train_funiegan.py