Automatic colorization

  1. Download generator and denoiser weights. Put generator and extractor weights in networks and denoiser weights in denoising/models.
  2. To colorize image or folder of images, use the following command:
$ python inference.py -p "path to file or folder"
Original Colorization