PyTorch implementation of a grayscale image colorizer using a deep CNN. Based on Colorful Image Colorization.
The following examples were trained with this repository as-is in 10 hours on a GTX1070 using the LHQ256 dataset.
For the below images:
- Top left: Original image
- Top right: Grayscale image
- Bottom left: Network output
- Bottom right: Network output with 50% saturation boost
