/pytorch-lightning-unet

A simple implementation of the famous UNet architecture in pytorch using pytorch lightning

Primary LanguageJupyter NotebookMIT LicenseMIT

PyTorch-Lightning based UNet