← all repositories

yu-changqian/TorchSeg

PyTorch-based modular reference implementation for semantic segmentation models with distributed training and multi-GPU support.

1.4k stars Python Computer VisionML Frameworks
TorchSeg
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

TorchSeg provides a fast, modular framework for implementing and training semantic segmentation algorithms in PyTorch. It includes pre-trained models (ResNet18/50/101) and supports popular segmentation architectures such as FCN, DFN, BiSeNet, PSPNet, and PSANet. The project offers distributed training capabilities for multi-GPU setups and benchmarks on datasets including PASCAL VOC 2012, Cityscapes, ADE20K, and CamVid.

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