readme.md

UPD. See the improved version.

Automatic colorization

1. Download generator, extractor and denoiser weights. Put generator and extractor weights in model and denoiser weights in denoising/models.
2. To colorize image, folder of images, .cbz or .cbr file, use the following command:

$ python inference.py -p "path to file or folder"

Manual colorization with color hints

1. Download colorizer and denoiser weights. Put colorizer weights in model and denoiser weights in denoising/models.
2. Run gunicorn server with:

$ ./run_drawing.sh

3. Open localhost:5000 with a browser.

References

1. Extractor weights are taken from https://github.com/blandocs/Tag2Pix/releases/download/release/model.pth
2. Denoiser weights are taken from http://www.ipol.im/pub/art/2019/231.