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

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.