/UNet-Pytorch

Simple PyTorch implementations of U-Net (with ResNet-50 encoder) for multi-class image segmentation with custom dataset.

Primary LanguagePython

Stargazers