pytorch-segmentation

Segmentation model library

Implementation of semantic segmentation models and datasets using PyTorch

art Semantic segmentation models, datasets and losses implemented in PyTorch.

GitHub

2k stars
22 watching
386 forks
Language: Jupyter Notebook
last commit: over 2 years ago
Linked from 2 awesome lists

computer-visiondeep-learningsemantic-segmentation

Backlinks from these awesome lists:

Related projects:

RepositoryDescriptionStars
zijundeng/pytorch-semantic-segmentationProvides PyTorch implementations of various models and pipelines for semantic segmentation in deep learning.1,729
kazuto1011/deeplab-pytorchPyTorch implementation of DeepLab v2 for semantic segmentation on COCO-Stuff and PASCAL VOC datasets1,098
davidtvs/pytorch-enetA PyTorch implementation of a real-time semantic segmentation model using ENet architecture392
thuyngch/human-segmentation-pytorchAn open source implementation of deep learning-based human segmentation models using PyTorch558
kazuto1011/pspnet-pytorchRe-implementation of a deep learning model for semantic segmentation using PyTorch.52
speedinghzl/pytorch-segmentation-toolboxProvides PyTorch implementations of deep learning models for semantic segmentation772
javeywang/pyramid-attention-networks-pytorchAn implementation of a deep learning model using PyTorch for semantic segmentation tasks.237
media-smart/vedasegA PyTorch-based toolbox for building and training semantic segmentation models408
tramac/fast-scnn-pytorchA PyTorch implementation of a deep learning model for semantic segmentation tasks in computer vision.380
eromera/erfnet_pytorchProvides a PyTorch implementation of the ERFNet architecture for semantic segmentation432
lextal/pspnet-pytorchA PyTorch implementation of a segmentation network architecture588
4uiiurz1/pytorch-nested-unetAn implementation of a deep learning model for image segmentation using PyTorch868
wizaron/reseg-pytorchA PyTorch implementation of a deep learning model for semantic image segmentation with annotated object parts.46
hszhao/semsegA PyTorch implementation of semantic segmentation models with support for multiprocessing training and various backbones.1,347
yunlongdong/fcn-pytorchA PyTorch implementation of FCN for semantic segmentation with an easy-to-use interface and pre-trained models.161