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
382 forks
Language: Jupyter Notebook
last commit: 8 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,724
kazuto1011/deeplab-pytorch PyTorch implementation of DeepLab v2 for semantic segmentation on COCO-Stuff and PASCAL VOC datasets 1,093
davidtvs/pytorch-enet A PyTorch implementation of a real-time semantic segmentation model using ENet architecture 389
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. 235
media-smart/vedaseg A PyTorch-based toolbox for building and training semantic segmentation models 410
tramac/fast-scnn-pytorch A PyTorch implementation of a deep learning model for semantic segmentation tasks in computer vision. 381
eromera/erfnet_pytorch Provides a PyTorch implementation of the ERFNet architecture for semantic segmentation 429
lextal/pspnet-pytorch A PyTorch implementation of a segmentation network architecture 585
4uiiurz1/pytorch-nested-unet An implementation of a deep learning model for image segmentation using PyTorch 858
wizaron/reseg-pytorch A PyTorch implementation of a deep learning model for semantic image segmentation with annotated object parts. 45
hszhao/semseg A PyTorch implementation of semantic segmentation models with support for multiprocessing training and various backbones. 1,343
yunlongdong/fcn-pytorch A PyTorch implementation of FCN for semantic segmentation with an easy-to-use interface and pre-trained models. 160