segmentation_models.pytorch
Image Segmentation Library
A PyTorch library for building and training neural networks for image segmentation tasks.
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
10k stars
82 watching
2k forks
Language: Python
last commit: 3 days 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,393 |
 4uiiurz1/pytorch-nested-unet | An implementation of a deep learning model for image segmentation using PyTorch | 851 |
 csailvision/semantic-segmentation-pytorch | An implementation of semantic segmentation models on the MIT ADE20K dataset using PyTorch. | 4,948 |
 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. | 381 |
 yassouali/pytorch-segmentation | Implementation of semantic segmentation models and datasets using PyTorch | 1,686 |
 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,428 |
 lextal/pspnet-pytorch | A PyTorch implementation of a segmentation network architecture | 585 |
 mattmacy/vnet.pytorch | A PyTorch implementation of V-Net for volumetric medical image segmentation | 694 |
 media-smart/vedaseg | A PyTorch-based toolbox for building and training semantic segmentation models | 410 |
 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,251 |
 davidtvs/pytorch-enet | A PyTorch implementation of a real-time semantic segmentation model using ENet architecture | 389 |
 javeywang/pyramid-attention-networks-pytorch | An implementation of a deep learning model using PyTorch for semantic segmentation tasks. | 235 |