Future work on this project Replace Bilinear upsampling method with tf.image.resize_BILINEAR See if SSIM impacts loss, if yes, how much? Try smaller network?