ML

A from-scratch PyTorch implementation of the convolutional neural network described by Zhang et al. in Colorful Image Colorization. Can be used with pretrained weights or trained from scratch. Also includes an implementation of Google’s DeepLab-V3+ network that can be substituted for the default VGG network.