Image Segmentation Python Packages

segmentation-models-pytorch

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

medpy

Medical image processing in Python

mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

kimimaro

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

pictograph

Complete python SDK, CLI, and API reference for integrating with the Pictograph.io platform.

yautja

Yautja is a sci-fi image segmentation, re-skinning, and annotation skill for claude, codex and more with thermal-imaging and HUD-style output. For entertainment purposes only; re-skinning colors are algorithmically generated, not measured temperatures. www.yautja.ai

pyclesperanto

GPU-accelerated Image Processing library for Python

segmentation-models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

catalyst

Accelerated deep learning R&D

smriprep

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

pytorch-toolbelt

PyTorch extensions for fast R&D prototyping and Kaggle farming

cleanvision

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

tinybrain

Image pyramid generation for grayscale and segmentation image resize.

img2num

Cross-platform library for converting natural raster images into clean SVGs - fast, precise and configurable.

awesomepmc

Copula-based pairwise Markov chains for non-Gaussian time-series classification, estimation, imputation and forecasting.

caer

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

rlemasklib

Manipulate run-length encoded image masks

napari-vipp

VIPP — Visual Image Processing Platform: Visual image processing made approachable

keras-unet

Helper package with multiple U-Net implementations in Keras as well as useful utility tools helpful when working with image semantic segmentation tasks. This library and underlying tools come from multiple projects I performed working on semantic segmentation tasks

otx

Build, train, optimize, and run computer vision models locally, from raw images to live inference. Open source, optimized for Intel XPU (CPU-only and CUDA also supported).

biapy

Open source Python library for building bioimage analysis pipelines

hbb2obb

🔄 Converts horizontal bounding boxes into oriented (rotated) boxes by prompting SAM-family models. Includes model ensembling, confidence scoring, evaluation, format conversion, and hyperparameter tuning.

mvtb-data

Machine vision toolbox for Python

ovkit

One line of Python for 20 AI capabilities on Intel CPU/GPU/NPU — detection, OCR, text-to-speech, pose. Offline, Apache-2.0, no AGPL.