skur42 / deepcolor

Automatic coloring and shading of manga-style lineart, using Tensorflow + cGANs

deepcolor

Automatic coloring and shading of manga-style lineart, using Tensorflow + cGANs

Read the writeup: http://kvfrans.com/coloring-and-shading-line-art-automatically-through-conditional-gans/

Try the demo: http://color.kvfrans.com

Setup

Prereqs

- Python 2.7, numpy
- Tensorflow 0.12
- OpenCV

Running it

1. make a folder called "results"
2. make a folder called "imgs"
3. Fill the "imgs" folder with your own .jpg images, or run "download_images.py" to download from Safebooru.
4. Run "python main.py train". I trained for ~20 epochs, taking about 16 hours on one GPU.
5. To sample, run "python main.py sample"
6. To start the server, run "python server.py". It will host on port 8000.

Pre-trained

Get the pretrained model: https://drive.google.com/file/d/0BydPPLNieijIdDlUYWxhelEwRnM/view?usp=sharing

Folder structure should go:

main.py
server.py
checkpoint/
    tr/
        checkpoint
        model-1101500.index
        model-1101500.data-00000-of-00001
        model-1101500.meta

Code based off this pix2pix implementation.