Semantic Segmentation Python Packages

ultralytics

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

nncf

Neural Network Compression Framework for enhanced OpenVINO™ inference

segmentation-models-pytorch

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

effdet

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

imgviz

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

cvat-sdk

Computer Vision Annotation Tool (CVAT) is a leading platform for building high-quality visual datasets for vision AI. It offers open-source, cloud, and enterprise products, as well as labeling services, for image, video, and 3D annotation with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

label-studio

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

gluoncv

Gluon CV Toolkit

mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

ultralytics-opencv-headless

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

segments-ai

Segments.ai Python SDK

kili

Simplest and fastest image and text annotation tool.

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.

cvat-cli

Computer Vision Annotation Tool (CVAT) is a leading platform for building high-quality visual datasets for vision AI. It offers open-source, cloud, and enterprise products, as well as labeling services, for image, video, and 3D annotation with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

pytorch-segmentation-models-trainer

Framework to train semantic segmentation models on Pytorch using yaml config files

fashn-human-parser

Human parsing model for fashion and virtual try-on applications

geodeep

Free and open source library for AI object detection and semantic segmentation in geospatial rasters. 🚀

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

groundino-samnet

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

biapy

Open source Python library for building bioimage analysis pipelines

rastervision-pytorch-learner

An open source library and framework for deep learning on satellite and aerial imagery.

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.

yolo-tiling

Tile (slice) a YOLO-formatted dataset for small detection, segmentation, and classification

rastervision

An open source library and framework for deep learning on satellite and aerial imagery.