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

kimimaro

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

catalyst

Accelerated deep learning R&D

segmentation-models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

pyclesperanto

GPU-accelerated Image Processing library for Python

smriprep

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

cleanvision

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

pytorch-toolbelt

PyTorch extensions for fast R&D prototyping and Kaggle farming

tinybrain

Image pyramid generation for grayscale and segmentation image resize.

rlemasklib

Manipulate run-length encoded image masks

caer

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

otx

Train, Evaluate, Optimize, Deploy Computer Vision Models via OpenVINO™

orthoseg

Package to make it easier to segment orthophotos.

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.

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.

samv2

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

biapy

Open source Python library for building bioimage analysis pipelines

napari-pyclesperanto-assistant

GPU-accelerated image processing in napari using OpenCL

machinevision-toolbox-python

Machine vision toolbox for Python

detectree

Tree detection from aerial imagery in Python

eiseg

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.