Awesome
Automatic colorization
- Download generator and denoiser weights. Put generator and extractor weights in
networks
and denoiser weights indenoising/models
. - To colorize image or folder of images, use the following command:
$ python inference.py -p "path to file or folder"
Original | Colorization |
---|---|
<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"> |