meetps/pytorch-semseg

Implements 8+ segmentation architectures (PSPNet, FCN, U-Net, SegNet, LinkNet, ICNet, FRRN) with pretrained weights and integrated support for 7 major datasets (Pascal VOC, Cityscapes, ADE20K, CamVid, NYUDv2, etc.). Provides YAML-based configuration for training with modular components including multiple optimizers, learning rate schedules, data augmentation pipelines, and loss functions. Includes utilities for validation with flip augmentation, inference on custom images with optional DenseCRF post-processing, and real-time FPS measurement.

3,411 stars. No commits in the last 6 months.