Colorization of Grayscale Images using Machine Learning

Setup

• Create virtual environment with Python 3.7:
  • Linux: $ python3.7 -m venv venv
  • Windows: $ py -3.7 -m venv venv
• Activate virtual environment
• Install requirements:
  • pip install -r requirements.txt
• Build Kernel with IPython for Jupyter Notebook:
  • ipython kernel install --user --name=Colorization
• Download training/testing set from: https://www.floydhub.com/emilwallner/datasets/colornet/2/images

Manual Version

Used Image: Photo by Christopher Campbell on Unsplash