UPD. See the improved version.
Automatic colorization
- Download generator, extractor and denoiser weights. Put generator and extractor weights in
model
and denoiser weights in denoising/models
.
- 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
- Download colorizer and denoiser weights. Put colorizer weights in
model
and denoiser weights in denoising/models
.
- Run gunicorn server with:
$ ./run_drawing.sh
- Open
localhost:5000
with a browser.
References
- Extractor weights are taken from https://github.com/blandocs/Tag2Pix/releases/download/release/model.pth
- Denoiser weights are taken from http://www.ipol.im/pub/art/2019/231.