UNet code in Pytorch. I downloaded the data from here: https://github.com/hoangp/isbi-datasets The code is slightly different from the UNet paper since I used padding to keep the sizes to be the same. I also used dropout after ReLU layers. The code was run using Google Colab and Kaggle notebook.