This is a keras implementation of paper Colorful Image Colorization.
Follow the instruction to download ImageNet dataset.
Extract training images:
$ python pre-process.py$ python train.pyIf you want to visualize during training, run in your terminal:
$ tensorboard --logdir path_to_current_dir/logsDownload pre-trained model into "models" folder then run:
$ python demo.py| Input | Output | GT |
|---|---|---|
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |
 |