/UResnet

Underwater Image Enhancement

Primary LanguagePython

UResnet

A method to to underwater image enhancement.
CycleGan is used to generate paired images to train UResnet.
[Paper]

Samples of Training Set

trainset

Samples of Model Outputs

results

Requirements

  • Linux (code have not been tested on Windows or OSX)
  • NVIDIA GPU + CUDA CuDNN
  • pytorch >= 0.4.1

Train model

use python train.py --input_images_path xxx/path/to/input/ --label_images_path xxx/path/to/label --snapshots_folder xxx/path/to/save/snapshot
use python train.py -h to choose more usable options

Dataset

The training set should be organized like
Input image: xxx/path/input/abcd.jpg Label image: xxx/path/label/abcd_label.jpg