PyTorch Image Segmentation Tutorial with U-NET: everything from scratch baby