segmentation_models.pytorch
Segmentation Model Library
A comprehensive library for training and applying deep learning models for image segmentation
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
10k stars
83 watching
2k forks
Language: Python
last commit: 11 months ago
Linked from 2 awesome lists
computer-visiondeeplab-v3-plusdeeplabv3fpnimage-processingimage-segmentationimagenetmodelspretrained-modelspretrained-weightspspnetpytorchsegformersegmentationsegmentation-modelssemantic-segmentationtransformersunetunet-pytorchunetplusplus
josephmisiti/awesome-machine-learning
mrgloom/awesome-semantic-segmentationRelated projects:
| Repository | Description | Stars |
|---|---|---|
 meetps/pytorch-semseg | Implementations of popular semantic segmentation architectures using PyTorch | 3,400 |
 4uiiurz1/pytorch-nested-unet | An implementation of a deep learning model for image segmentation using PyTorch | 868 |
 csailvision/semantic-segmentation-pytorch | An implementation of semantic segmentation models on the MIT ADE20K dataset using PyTorch. | 4,963 |
 speedinghzl/pytorch-segmentation-toolbox | Provides PyTorch implementations of deep learning models for semantic segmentation | 772 |
 thuyngch/human-segmentation-pytorch | An open source implementation of deep learning-based human segmentation models using PyTorch | 558 |
 tramac/fast-scnn-pytorch | A PyTorch implementation of a deep learning model for semantic segmentation tasks in computer vision. | 380 |
 yassouali/pytorch-segmentation | Implementation of semantic segmentation models and datasets using PyTorch | 1,705 |
 oyam/pytorch-dpns | PyTorch implementation of a deep learning model for image segmentation | 90 |
 pytorch/examples | A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. | 22,530 |
 lextal/pspnet-pytorch | A PyTorch implementation of a segmentation network architecture | 588 |
 mattmacy/vnet.pytorch | A PyTorch implementation of V-Net for volumetric medical image segmentation | 703 |
 media-smart/vedaseg | A PyTorch-based toolbox for building and training semantic segmentation models | 408 |
 yunjey/pytorch-tutorial | A comprehensive tutorial project that provides code examples for learning PyTorch by implementing various deep learning models and demonstrating their usage. | 30,401 |
 davidtvs/pytorch-enet | A PyTorch implementation of a real-time semantic segmentation model using ENet architecture | 392 |
 javeywang/pyramid-attention-networks-pytorch | An implementation of a deep learning model using PyTorch for semantic segmentation tasks. | 237 |