Adding Colour To B&W Images Using GANs and U-Nets
Using PyTorch and Deep Learning applications, this project implements U-Nets and Feature Loss Applications inspired by the paper “Perceptual Losses for Real-Time Style Transfer and Super-Resolution” by Johnson et al to determine the color of black and white images.
This project is then improved using U-Nets and Feature Loss Applications inspired by the paper “Perceptual Losses for Real-Time Style Transfer and Super-Resolution” by Johnson et al. Instead of using MSELoss on pixel values and a CrossEntropy loss for the Critic model, we compare features displayed in the real image and the generated image.
Johnson, et al. “Perceptual Losses for Real-Time Style Transfer and Super-Resolution.” Computer Vision – ECCV 2016 Lecture Notes in Computer Science, 2016, pp. 694–711.
Daniel Diamond
data watches music travel