Image Segmentation Python Packages

segmentation-models-pytorch

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

mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

medpy

Medical image processing in Python

catalyst

Accelerated deep learning R&D

segmentation-models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

kimimaro

Skeletonize densely labeled 3D image segmentations with TEASAR. (Medial Axis Transform)

smriprep

Structural MRI PREProcessing (sMRIPrep) workflows for NIPreps (NeuroImaging PREProcessing tools)

cleanvision

Automatically find issues in image datasets and practice data-centric computer vision.

pyclesperanto

GPU-accelerated Image Processing library for Python

tinybrain

Image pyramid generation for grayscale and segmentation image resize.

pytorch-toolbelt

PyTorch extensions for fast R&D prototyping and Kaggle farming

caer

High-performance Vision library in Python. Scale your research, not boilerplate.

rlemasklib

Manipulate run-length-encoded image masks

pottslab

Unsupervised multilabel image segmentation (color/gray/multichannel) based on the Potts model (aka piecewise constant Mumford-Shah model)

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.

paddleseg

Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image Matting, 3D Segmentation, etc.

otx

OpenVINO™ Training Extensions: Train, Evaluate, Optimize, Deploy Computer Vision Models via OpenVINO™

machinevision-toolbox-python

Machine vision toolbox for Python

biapy

Open source Python library for building bioimage analysis pipelines

orthoseg

Train your own neural networks to segment orthophotos.

detectree

Tree detection from aerial imagery in Python.

samv2

CPU compatible fork of the official SAMv2 implementation aimed at more accessible and documented tutorials

particleanalyzer

SEM Image Analysis Tool