Semantic Segmentation Python Packages

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

nncf

Neural Network Compression Framework for enhanced OpenVINO™ inference

label-studio

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

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.

gluoncv

Gluon CV Toolkit

mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

segments-ai

Segments.ai Python SDK

kili

Simplest and fastest image and text annotation tool.

atomai

Deep and Machine Learning for Microscopy

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.

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.

rastervision

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

labelme2coco

A lightweight package for converting your labelme annotations into COCO object detection format.

rastervision-pytorch-learner

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

rastervision-core

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

rastervision-pytorch-backend

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

rastervision-pipeline

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

rastervision-aws-s3

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

rastervision-aws-batch

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

vernier

A modern toolkit to evaluate 2D detection and segmentation models

pytorch-segmentation-models-trainer

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

geodeep

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