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