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

gluoncv

Gluon CV Toolkit

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

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

segments-ai

Segments.ai Python SDK

kili

Simplest and fastest image and text annotation tool.

cvat-cli

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

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.

geodeep

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

rastervision-core

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

rastervision-pytorch-learner

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

A rastervision plugin that adds an AWS S3 file system

rastervision-pytorch-backend

A rastervision plugin that adds PyTorch backends for rastervision.core pipelines

rastervision-aws-batch

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

yolo-tiling

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

fashn-human-parser

Human parsing model for fashion and virtual try-on applications

groundino-samnet

A SAM model with GroundingDINO model for feet segmentation

pytorch-segmentation-models-trainer

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