Repository Index

Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

• ultralytics
• yolov8
• yolo
• deep-learning

OpenMMLab Detection Toolbox and Benchmark

• object-detection
• instance-segmentation
• fast-rcnn
• faster-rcnn

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

• segmentation
• image-processing
• pspnet
• unet

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

• pytorch
• unet
• pytorch-unet
• semantic-segmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

• semantic-segmentation
• pytorch
• pspnet
• deeplabv3

Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image Matting, 3D Segmentation, etc.

• semantic-segmentation
• image-segmentation
• image-matting
• interactive-segmentation

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

• pytorch
• semantic-segmentation
• scene-recognition
• ade20k

Official PyTorch implementation of SegFormer

• semantic-segmentation
• transformer
• ade20k
• cityscapes

Semantic Segmentation Architectures Implemented in PyTorch

• pytorch
• semantic-segmentation
• deep-learning
• fully-convolutional-networks

The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919

• segmentation
• semantic-segmentation
• cityscapes
• pascal-context

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

• semantic-segmentation
• pytorch

[ECCV 2024] Official implementation of the paper "Semantic-SAM: Segment and Recognize Anything at Any Granularity"

[IEEE TMI Best Paper Award] Official Implementation for UNet++

• medical-imaging
• segmentation
• unet
• biomedical-imaging

Using modified BiSeNet for face parsing in PyTorch

• face-parsing
• semantic-segmentation
• pytorch
• celeba-hq-dataset

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

• semantic-segmentation
• unet
• unet-pytorch
• 3d-unet

An extension of Open3D to address 3D Machine Learning tasks

• 3d-perception
• datasets
• pretrained-models
• lidar

Automated dense category annotation engine that serves as the initial semantic labeling for the Segment Anything dataset (SA-1B).

Official Pytorch implementation of the preprint paper "Castle in the Sky: Dynamic Sky Replacement and Harmonization in Videos", in arXiv:2010.11800.

• dynamic-sky-replacement
• sky-matting
• harmonization
• skyar

Nvidia Semantic Segmentation monorepo

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

• semantic-segmentation
• computer-vision
• deep-learning

[CVPR 2023] OneFormer: One Transformer to Rule Universal Image Segmentation

• ade20k
• cityscapes
• coco
• image-segmentation

All-in-one training for vision models (YOLO, ViTs, RT-DETR, DINOv3): pretraining, fine-tuning, distillation.

• computer-vision
• contrastive-learning
• deep-learning
• distillation

PyTorch for Semantic Segmentation

• semantic-segmentation
• deep-learning
• fully-convolutional-networks
• pytorch

[CVPR 2023] Official implementation of the paper "Mask DINO: Towards A Unified Transformer-based Framework for Object Detection and Segmentation"

• instance-segmentation
• object-detection
• panoptic-segmentation
• semantic-segmentation