🎨 Semantic segmentation models, datasets and losses implemented in PyTorch.

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.

UNet model with VGG11 encoder pre-trained on Kaggle Carvana dataset

PyTorch for Semantic Segmentation