← 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
Not currently ranked — collecting fresh signals.
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.

Frequently asked

What is qubvel/segmentation_models?
A Keras/TensorFlow library for image segmentation offering pretrained Unet, FPN, Linknet, and PSPNet architectures with 25 backbone options.
Is segmentation_models open source?
Yes — qubvel/segmentation_models is open source, released under the MIT license.
What language is segmentation_models written in?
qubvel/segmentation_models is primarily written in Python.
How popular is segmentation_models?
qubvel/segmentation_models has 4.9k stars on GitHub.
Where can I find segmentation_models?
qubvel/segmentation_models is on GitHub at https://github.com/qubvel/segmentation_models.

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