EnahnceNet: LiteEnhanceNet: A Lightweight Network for Real�time Single Underwater Image Enhancement

Train the Model

python training.py

python test.py

python=3.8 pytorch=1.11 cudnn=8.2 numpy=1.22

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
data		data
snapshots		snapshots
utils		utils
.gitattributes		.gitattributes
SSIM_loss.py		SSIM_loss.py
VGG_loss.py		VGG_loss.py
combined_loss.py		combined_loss.py
dataloader.py		dataloader.py
metrics_calculation.py		metrics_calculation.py
model.py		model.py
readme.md		readme.md
test.py		test.py
training.py		training.py
uiqm_utils.py		uiqm_utils.py