qubvel/segmentation_models
A Keras/TensorFlow library for image segmentation offering pretrained Unet, FPN, Linknet, and PSPNet architectures with 25 backbone options.

Velocity · 7d
+1.7
★ / day
Trend
→steady
star history
This library provides high-level APIs for building neural network models for binary and multi-class image segmentation. It implements four major segmentation architectures with 25 available pretrained backbones including ResNet, EfficientNet, DenseNet, and MobileNet. The package includes segmentation-specific losses (Jaccard, Dice, Focal) and metrics (IoU, F-score) for training and evaluation.