Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset

Semantic Segmentation Architectures Implemented in PyTorch

Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)

3D U-Net model for volumetric semantic segmentation written in pytorch

pytorch implementation for "PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation" https://arxiv.org/abs/1612.00593

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet

Pytorch 3D U-Net Convolution Neural Network (CNN) designed for medical image segmentation

A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation

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

PyTorch for Semantic Segmentation

Fast, modular reference implementation and easy training of Semantic Segmentation algorithms in PyTorch.

Semantic Segmentation in Pytorch

Official PyTorch implementation of Superpoint Transformer introduced in [ICCV'23] "Efficient 3D Semantic Segmentation with Superpoint Transformer" and SuperCluster introduced in [3DV'24 Oral] "Scalable 3D Panoptic Segmentation As Superpoint Graph Clustering"

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help

SOTA Semantic Segmentation Models in PyTorch

Official PyTorch implementation of ODISE: Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Models [CVPR 2023 Highlight]

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.

A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.

[ICCV2021] Official PyTorch implementation of Segmenter: Transformer for Semantic Segmentation

Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation

SSSegmentation: An Open Source Supervised Semantic Segmentation Toolbox Based on PyTorch.

Official Pytorch implementations for "SegNeXt: Rethinking Convolutional Attention Design for Semantic Segmentation" (NeurIPS 2022)

Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021

Official PyTorch implementation of GroupViT: Semantic Segmentation Emerges from Text Supervision, CVPR 2022.

PyTorch Implementations for DeeplabV3 and PSPNet

U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI

Image Segmentation and Object Detection in Pytorch