This is a Deep Learning academic project, throughout this project we explored state of the art deep learning techniques for automatic colorization of gray scale images. For further details check this report.
- DDColor
- Conditional Generative Adversial Networks (cGANS)
- PyTorch
- tqdm
- pandas
- dotenv
- skimage
- PIL
- numpy
The code expects the COCO-2017 dataset stored in the coco-2017-dataset directory . You can set the path for this and other directories in a .env file.
Follow the structure of .env.template