Segmentation Models
Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
model-store models unet computer-vision segmentation code keras tensorflow library

The main features of this library are:

  • High level API (just two lines of code to create model for segmentation)
  • 4 models architectures for binary and multi-class image segmentation (including legendary Unet)
  • 25 available backbones for each architecture
  • All backbones have pre-trained weights for faster and better convergence
  • Helpful segmentation losses (Jaccard, Dice, Focal) and metrics (IoU, F-score)

Don't forget to tag @qubvel in your comment, otherwise they may not be notified.

Authors community post
Share this project
Similar projects
TensorFlow Model Garden
Models and examples built with TensorFlow.