Semantic Segmentation Python Packages

ultralytics

Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

effdet

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights

nncf

Neural Network Compression Framework for enhanced OpenVINO™ inference

segmentation-models-pytorch

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

imgviz

Rich Image Visualization with Minimum Dependency (no OpenCV, Matplotlib)

cvat-sdk

CVAT is a leading data annotation platform for image, video, audio, and 3D datasets. It offers open-source, cloud, and enterprise products, as well as labeling services, with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

gluoncv

Gluon CV Toolkit

mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

vernier

A modern toolkit to evaluate 2D detection and segmentation models

segments-ai

Segments.ai Python SDK

label-studio

Label Studio is a multi-type data labeling and annotation tool with standardized output format

ultralytics-opencv-headless

Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

kili

Simplest and fastest image and text annotation tool.

cvat-cli

CVAT is a leading data annotation platform for image, video, audio, and 3D datasets. It offers open-source, cloud, and enterprise products, as well as labeling services, with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

fashn-human-parser

Human parsing model for fashion and virtual try-on applications

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

omniwatermask

Python library for high-accuracy water segmentation in satellite and aerial imagery, combining deep learning with NDWI and vector data for robust detection across multiple sensors and resolutions.

biapy

Open source Python library for building bioimage analysis pipelines

atomai

Deep and Machine Learning for Microscopy

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.

chainercv2

Sandbox for training deep learning networks

fiducio

Post-hoc calibration for 2D/3D semantic segmentation in PyTorch: temperature, vector, matrix and Dirichlet scaling, translation-invariant variants, and decision-preserving calibrators. Official code of "Rethinking Post-Hoc Calibration in Semantic Segmentation" (TMLR)

groundino-samnet

Workflow for foot detection and segmentation in patients in labor using GroundingDINO to generate bounding boxes and SAM for precise segmentation.

mapcv

Turn imagery and labels into ready-to-train remote-sensing datasets, fast.