We adapted the U-Net structure from: https://github.com/milesial/Pytorch-UNet
Adapted code from ECE324 Lecture 27 - "Autoencoders"
We adapted the U-Net structure from: https://github.com/milesial/Pytorch-UNet
Adapted code from ECE324 Lecture 27 - "Autoencoders"