Segmentation Models Python Packages

segmentation-models-pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

segmentation-models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

totalspineseg

Robust Segmentation and Labeling of Vertebrae, Intervertebral Discs, Spinal Cord, and Spinal Canal in MRI Images Using nnU-Net and Iterative Algorithm.

segmentation-models-pytorch-3d

Segmentation models for 3D data with different backbones. PyTorch.

quadra

Quadra: Effortless and reproducible deep learning workflows with configuration files.

d3m-segmentation-models-pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

saber-em

Segment Organelles from Electron Microscopy Datasets with Segment-Anything 2

segmentation-models-pytorch-deepflash2

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

segmentation-models-zwy

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

anomalib-orobix

Orobix anomalib fork

torch-segment

A deep learning image segmentation library and API on top of PyTorch.

geotorchai

GeoTorchAI: A Framework for Training and Using Spatiotemporal Deep Learning Models at Scale

rembg-greenscreen

Rembg Virtual Green Screen

segmentation-sust

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.