Colorful Image Colorization

This is a keras implementation of paper Colorful Image Colorization.

Dependencies

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

Name		Name	Last commit message	Last commit date
Latest commit History 107 Commits
data		data
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
cielab_color_space.py		cielab_color_space.py
class_rebal.py		class_rebal.py
config.py		config.py
data_generator.py		data_generator.py
demo.py		demo.py
demo_names.txt		demo_names.txt
generate_demo_file_names.py		generate_demo_file_names.py
generate_file_names.py		generate_file_names.py
model.py		model.py
run_demo.py		run_demo.py
train.py		train.py
train_names.txt		train_names.txt
unit_tests.py		unit_tests.py
utils.py		utils.py
valid_names.txt		valid_names.txt