U-Net semantic segmentation in PyTorch. It has a custom N-class one-hot encoder and supports aerial image datasets. 12 stars, 4 forks.
Source: github.com/madhavkhoslaa/Pytorch-U-Net-Segmentation (External link)
Project
Project
U-Net semantic segmentation in PyTorch. It has a custom N-class one-hot encoder and supports aerial image datasets. 12 stars, 4 forks.
Source: github.com/madhavkhoslaa/Pytorch-U-Net-Segmentation (External link)