← all repositories

qubvel-org/segmentation_models.pytorch

A PyTorch library for semantic image segmentation using encoder-decoder neural network architectures.

11.6k stars Python Computer VisionML Frameworks
segmentation_models.pytorch
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.