← all repositories

CSAILVision/semantic-segmentation-pytorch

A PyTorch implementation of semantic segmentation models trained on the MIT ADE20K scene parsing dataset.

5.1k stars Python Computer VisionML Frameworks
semantic-segmentation-pytorch
Velocity · 7d
+1.6
★ / day
Trend
steady
star history

This repository provides PyTorch implementations of semantic segmentation models for scene parsing on the MIT ADE20K dataset, which is the largest open-source dataset for semantic segmentation. The code includes training pipelines, pre-trained model weights, and a synchronized batch normalization module that computes statistics across multiple devices for improved segmentation accuracy. Users can run inference via a web demo, a Colab notebook, or by loading models programmatically.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.