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: 9 months ago
Linked from 2 awesome lists

computer-visiondeep-learningsemantic-segmentation

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
zijundeng/pytorch-semantic-segmentation Provides PyTorch implementations of various models and pipelines for semantic segmentation in deep learning. 1,729
kazuto1011/deeplab-pytorch PyTorch implementation of DeepLab v2 for semantic segmentation on COCO-Stuff and PASCAL VOC datasets 1,098
davidtvs/pytorch-enet A PyTorch implementation of a real-time semantic segmentation model using ENet architecture 392
thuyngch/human-segmentation-pytorch An open source implementation of deep learning-based human segmentation models using PyTorch 558
kazuto1011/pspnet-pytorch Re-implementation of a deep learning model for semantic segmentation using PyTorch. 52
speedinghzl/pytorch-segmentation-toolbox Provides PyTorch implementations of deep learning models for semantic segmentation 772
javeywang/pyramid-attention-networks-pytorch An implementation of a deep learning model using PyTorch for semantic segmentation tasks. 237
media-smart/vedaseg A PyTorch-based toolbox for building and training semantic segmentation models 408
tramac/fast-scnn-pytorch A PyTorch implementation of a deep learning model for semantic segmentation tasks in computer vision. 380
eromera/erfnet_pytorch Provides a PyTorch implementation of the ERFNet architecture for semantic segmentation 432
lextal/pspnet-pytorch A PyTorch implementation of a segmentation network architecture 588
4uiiurz1/pytorch-nested-unet An implementation of a deep learning model for image segmentation using PyTorch 868
wizaron/reseg-pytorch A PyTorch implementation of a deep learning model for semantic image segmentation with annotated object parts. 46
hszhao/semseg A PyTorch implementation of semantic segmentation models with support for multiprocessing training and various backbones. 1,347
yunlongdong/fcn-pytorch A PyTorch implementation of FCN for semantic segmentation with an easy-to-use interface and pre-trained models. 161