-
models/
consists the cGAN models & visualizations.- run
train.py
to train on images stored indatasets/train
. - run
test.py
to generate colorization for images indatasets/test
. datasets/generated_Images
contains generated images and trained model weights.datasets/test
contains testing images.datasets/train
contains/put dataset for training.datasets/validation
contains validation set of images for training.
- run
-
Segmentation and post-processing
segmentation/input_images
contains test imagessegmentation/bw_images
contains the black and white images after screentone removalResults
contains results for postprocessing and final outputs- run
segmentation.m
in MATLAB to segment the input image (Monochrome Original)
- a - input: monochrome original
- b - output: cGAN output
- c - output: Post Processed output