Image Segmentation Python Packages

segmentation-models-pytorch

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

rlemasklib

Manipulate run-length encoded image masks

mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

catalyst

Accelerated deep learning R&D

medpy

Medical image processing in Python

kimimaro

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

segmentation-models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

cleanvision

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

tinybrain

Image pyramid generation for grayscale and segmentation image resize.

pyclesperanto

GPU-accelerated Image Processing library for Python

pytorch-toolbelt

PyTorch extensions for fast R&D prototyping and Kaggle farming

smriprep

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

doxapy

A Local Adaptive Thresholding framework for image binarization written in C++, with JS, Python and MATLAB bindings. Implementing: Otsu, Bernsen, Niblack, Bradley, Sauvola, Wolf, Gatos, NICK, Su, T.R. Singh, WAN, ISauvola, Feng, Phansalkar, AdOtsu, along with DRDM and all Pseudo Merics.

caer

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

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

biapy

Open source Python library for building bioimage analysis pipelines

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

Build computer vision models in a fraction of the time and with less data.

occlude

Blur immodestly dressed people in videos using Islamic modesty rules. A command-line tool: video in, clean video out, original audio preserved. Built on YOLOv8, SegFormer, and InsightFace.

machinevision-toolbox-python

Machine vision toolbox for Python

skin-tone-classifier

An easy-to-use library for skin tone classification

napari-pyclesperanto-assistant

GPU-accelerated general purpose image processing and image analysis in napari

detectree

Tree detection from aerial imagery in Python

mist-medical

MIST: A simple and scalable end-to-end framework for 3D medical imaging segmentation.