UNet adapted for SuperResolution, implemented as homework for Artificial Intelligence II at CIMAT. The model uses downscaled STL-10 patches to for training (96x96 to 48x48)
The script unetSR.py contains the class with the UNet implementation.
The notebook UnetSR.ipynb contains the training the net and an inference example using the trained weights (unet_final_1.pt).