← all repositories

qubvel/segmentation_models

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

4.9k stars Python Computer VisionML Frameworks
segmentation_models
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.

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