config.py: config variables like batch size, training_iters and so on
image_helper.py: all functions related to image manipulation
read_input.py: all functions related to input
residual_encoder.py: the residual encoder model
common.py: the common part for training and testing, which is mainly the workflow for this model
train.py: train the residual encoder model using TensorFlow built-in AdamOptimizer
test.py: test your own images and save the output images
Download pre-trained VGG16 model vgg16.npy to vgg folder
Option 1: Use pre-trained residual encoder model
summary_path(you can change this path in
Option 2: Train your own model!
training_itersif you want.
training_dirto your directory that has all your training jpg images
testing_dirto your directory that has all your testing jpg images
More example output images can be found in sample_output_images folder.
GNU GPL 3.0 for personal or research use. COMMERCIAL USE PROHIBITED.