UNET-pytorch_implementation

This repository has the code of my implementation of the UNET architecture for semantic segmentation in Pytorch

How to use

1. Clone the repo
2. Run the .ipynb file
3. The model will be trained and can be used to train on 3 channel and 1 channel images of variable shapes.