qubvel-org/segmentation_models.pytorch
A PyTorch library for semantic image segmentation using encoder-decoder neural network architectures.

Velocity · 7d
+4.4
★ / day
Trend
→steady
star history
This library provides a high-level API for creating semantic segmentation models in PyTorch. It includes 12 model architectures such as Unet, Unet++, Segformer, DPT, FPN, and PSPNet, paired with over 800 pretrained convolution- and transformer-based encoders from timm. The library also bundles common training utilities including metrics and loss functions for segmentation tasks.