Automatic coloring and shading of manga-style lineart, using Tensorflow + cGANs
Python HTML JavaScript CSS
Switch branches/tags
Nothing to show
Clone or download
Permalink
Failed to load latest commit information.
web fixed sampling Mar 12, 2017
.gitignore nice Mar 5, 2017
README.md fixes Mar 5, 2017
download_images.py fixed sampling Mar 12, 2017
guess_colors.py reworked color prediction in VAE Mar 14, 2017
main.py fixed sampling Mar 12, 2017
server.py reworked color prediction in VAE Mar 14, 2017
test_processing.py Auto color! Mar 5, 2017
utils.py reworked color prediction in VAE Mar 14, 2017

README.md

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.