GitHub - sboominathan/colorization-tf: A Tensorflow implementation of ECCV2016 paper([Colorful Image Colorization](https://arxiv.org/pdf/1603.08511.pdf))

#Colorization By nilboy

A Tensorflow implementation of ECCV2016 paper(Colorful Image Colorization)

###Train

Download imagenet data and Extract to one directory named 'Imagenet'
link the Imagenet data directory to this project path
```
ln -s $Imagenet data/imagenet
```

python tools/create_imagenet_list.py

python tools/train.py -c conf/train.cfg

transform your training data to text_record file
calculate the training data prior-probs(reference to tools/create_prior_probs.py)
write your own train-configure file
train (python tools/train.py -c $your_configure_file)

###test demo

Download pretrained model(https://drive.google.com/file/d/0B-yiAeTLLamRWVVDQ1VmZ3BxWG8/view?usp=sharing)
```
mv color_model.ckpt models/model.ckpt
```
Test
```
python demo.py
```

Name		Name	Last commit message	Last commit date
Latest commit History 21 Commits
colorized_images		colorized_images
conf		conf
data		data
resources		resources
tools		tools
.gitignore		.gitignore
README.md		README.md
__init__.py		__init__.py
data.py		data.py
demo.py		demo.py
eval_tuner.py		eval_tuner.py
gray.jpg		gray.jpg
net.py		net.py
ops.py		ops.py
resnet.py		resnet.py
rgb_error.py		rgb_error.py
rgb_tuner.py		rgb_tuner.py
solver.py		solver.py
solver_multigpu.py		solver_multigpu.py
train_tuner.py		train_tuner.py
utils.py		utils.py

sboominathan/colorization-tf