qubvel-org/segmentation_models.pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
Python11,746 stars1,842 forksMIT computer-visiondeeplab-v3-plusdeeplabv3dptfpnimage-processingimage-segmentationimagenetmodelspretrained-weightspspnetpytorchsegformersegmentationsegmentation-modelssemantic-segmentationtransformersunetunet-pytorchunetplusplus