segmentation_models.pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
A Python library for training AI models that identify exactly which pixels in an image belong to each object or category. Offers 12 architectures and over 800 pretrained backbones, all usable in two lines of code.
Python
★ 12k
2d ago
Explain →