qubvel/segmentation_models

This is stars and forks stats for /qubvel/segmentation_models repository. As of 27 Apr, 2024 this repository has 4434 stars and 1007 forks.

Python library with Neural Networks for Image Segmentation based on Keras and TensorFlow. 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...