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

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.