Home

Awesome

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"
OriginalColorization
<img src="figures/bw1.jpg" width="512"><img src="figures/color1.png" width="512">
<img src="figures/bw2.jpg" width="512"><img src="figures/color2.png" width="512">
<img src="figures/bw3.jpg" width="512"><img src="figures/color3.png" width="512">
<img src="figures/bw4.jpg" width="512"><img src="figures/color4.png" width="512">
<img src="figures/bw5.jpg" width="512"><img src="figures/color5.png" width="512">
<img src="figures/bw6.jpg" width="512"><img src="figures/color6.png" width="512">