-
models/consists the cGAN models & visualizations.- run
train.pyto train on images stored indatasets/train. - run
test.pyto generate colorization for images indatasets/test. datasets/generated_Imagescontains generated images and trained model weights.datasets/testcontains testing images.datasets/traincontains/put dataset for training.datasets/validationcontains validation set of images for training.
- run
-
Segmentation and post-processing
segmentation/input_imagescontains test imagessegmentation/bw_imagescontains the black and white images after screentone removalResultscontains results for postprocessing and final outputs- run
segmentation.min MATLAB to segment the input image (Monochrome Original)
- a - input: monochrome original
- b - output: cGAN output
- c - output: Post Processed output




