qweasdd / manga-colorization

24 stars 7 forks source link

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.