UResnet

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

Samples of Training Set

Samples of Model Outputs

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