This repository has the code of my implementation of the UNET architecture for semantic segmentation in Pytorch
- Clone the repo
- Run the .ipynb file
- The model will be trained and can be used to train on 3 channel and 1 channel images of variable shapes.