PyTorch implementation for semantic segmentation (DeepLabV3+, UNet, etc.) - View it on GitHub