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.

quadra

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

anomalib-orobix

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

segmentation-models-pytorch-3d

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

saber-em

Segment Organelles from Electron Microscopy Datasets with Segment-Anything 2

rembg-greenscreen

Rembg Video Virtual Green Screen Edition

segmentation-models-pytorch-deepflash2

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

d3m-segmentation-models-pytorch

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

geotorchai

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

torch-segment

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

segmentation-models-zwy

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

segmentation-sust

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