References Models articles: U-Net: Convolutional Networks for Biomedical Image Segmentation Understanding Semantic Segmentation with UNET Models implementations: U-Net for brain segmentation U-Net: Training Image Segmentation Models in PyTorch Pytorch documentation: Training a classifier Developing custom PyTorch dataloaders Visualization utilities Torchvision transforms Other Python libraries: tqdm Other articles: Early Stopping with PyTorch to Restrain your Model from Overfitting How do I split a custom dataset into training and test datasets?