This repository is not active
tuhinnn-py/Pytorch-Image-Colourisation-using-RGB-predictor
A finetuned U-Net model trained on the Modified Celeb-A dataset(containing face masks) for solving the problem of image colorisation. The model outputs normalized RGB values which is in contrast to the benchmark model which outputs LAB values.
Jupyter Notebook