Skip to content
/ segmentation_models.pytorch Public

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

smp.readthedocs.io/

License

MIT license
10.5k stars 1.7k forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

qubvel-org/segmentation_models.pytorch

Error
Looks like something went wrong!

About

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

smp.readthedocs.io/

Topics

computer-vision models image-processing transformers pytorch imagenet segmentation image-segmentation unet dpt semantic-segmentation pretrained-weights pspnet fpn deeplabv3 unet-pytorch deeplab-v3-plus segmentation-models unetplusplus segformer

Resources

Readme

License

MIT license
Activity
Custom properties

Stars

10.5k stars

Watchers

84 watching

Forks

1.7k forks
Report repository

Releases 15

Segmentation Models - v0.5.0 Latest
Apr 17, 2025
+ 14 releases

Sponsor this project

Packages

No packages published

Contributors 67

+ 53 contributors

Languages