← all repositories

Tramac/awesome-semantic-segmentation-pytorch

A reference implementation of semantic segmentation models (FCN, DeepLab, PSPNet, etc.) in PyTorch with training and evaluation scripts.

3.1k stars Python Computer Vision
awesome-semantic-segmentation-pytorch
Velocity · 7d
+1.2
★ / day
Trend
steady
star history

This project provides concise, modifiable implementations of semantic segmentation models using PyTorch, including FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, and other architectures. It offers training, evaluation, and demo scripts for single and multi-GPU setups with support for common datasets like Pascal VOC. Users can quickly train models by specifying model type, backbone, dataset, learning rate, and epochs.

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