Skip to content

CV-Reimplementation/Global-structure-guided-learning-framework-for-underwater-image-enhancement

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Environment

  • python == 3.7
  • torchvision == 0.7.0
  • pytorch == 1.6.0
  • cudatoolkit == 10.2

Training

Please prepare your dataset and put the input image into ./dataset/trainA, the reference image into ./dataset/trainB.

python scripts/script.py --train

Testing

python scripts/script.py --test

Reach me

E-mail: junko.lin@yahoo.com

About

Code of Global structure-guided learning framework for underwater image enhancement

Resources

Stars

Watchers

Forks

Languages