segmentation_models.pytorch

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

GitHub
9k stars
80 watching
2k forks
Language: Python
last commit: 1 day ago
Linked from 2 awesome lists

computer-visiondeeplab-v3-plusdeeplabv3fpnimage-processingimage-segmentationimagenetmodelspretrained-modelspretrained-weightspspnetpytorchsegformersegmentationsegmentation-modelssemantic-segmentationtransformersunetunet-pytorchunetplusplus

Screenshot of qubvel-org/segmentation_models.pytorch website

https://smp.readthedocs.io/

Backlinks from these awesome lists: