Image colorization

Tested on tiny-imagenet. Using mse_loss, trying to make the loss function from 'Colorful Image Colorization' to work properly but returns gray predictions as is.

References

• Model based on:

  • Zhang, Richard, Phillip Isola, and Alexei A. Efros. ‘Colorful Image Colorization’
    
    • Repo: Richzhang/colorization
      

• Color Quantization:

  • Brun, Luc, and Alain Trémeau. ‘Color Quantization’, n.d.
    
  • Puzicha et al. ‘On Spatial Quantization of Color Images’, 1998, 25.
    

• Distillation:

  • Hinton, Geoffrey et al. ‘Distilling the Knowledge in a Neural Network’