← all repositories

yassouali/pytorch-segmentation

A PyTorch library providing implementations of semantic segmentation models, datasets, losses, and training utilities.

1.8k stars Jupyter Notebook Computer VisionML Frameworks
pytorch-segmentation
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

This repository offers a collection of semantic segmentation models including Deeplab V3+ and GCN, implemented in PyTorch. It provides ready-to-use dataset loaders, various loss functions, learning rate schedulers, and data augmentation techniques. The project supports training and inference workflows through JSON config files and includes Colab notebook tutorials.

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