This is a keras implementation of paper Colorful Image Colorization.

• NumPy
• Tensorflow
• Keras
• OpenCV

Follow the instruction to download ImageNet dataset.

Extract training images:

$ python pre-process.py

$ python train.py

If you want to visualize during training, run in your terminal:

$ tensorboard --logdir path_to_current_dir/logs

Download pre-trained model into "models" folder then run:

$ python demo.py

Input	Output	GT