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.

d3m-segmentation-models-pytorch

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

segmentation-models-pytorch-deepflash2

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

segmentation-models-pytorch-3d

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

quadra

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

segmentation-models-zwy

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

anomalib-orobix

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

torch-segment

A library for deep learning image segmentation using PyTorch

geotorchai

GeoTorchAI, formarly GeoTorch, A Spatiotemporal Deep Learning Framework

saber-em

Segment Organelles from Electron Microscopy Datasets with Segment-Anything 2

rembg-greenscreen

Rembg Video Virtual Green Screen Edition

segmentation-sust

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