Machine Vision

State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow

Fast, flexible, and advanced augmentation library for deep learning, computer vision, and medical imaging. Albumentations offers a wide range of transformations for both 2D (images, masks, bboxes, keypoints) and 3D (volumes, volumetric masks, keypoints) data, with optimized performance and seamless integration into ML workflows.

Ultralytics YOLO 🚀 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.

John Snow Labs Spark NLP is a natural language processing library built on top of Apache Spark ML. It provides simple, performant & accurate NLP annotations for machine learning pipelines, that scale easily in a distributed environment.

A set of easy-to-use utils that will come in handy in any Computer Vision project

Image augmentation library for deep neural networks

QUick and DIrty Domain Adaptation

A toolkit for making real world machine learning and data analysis applications

Light Weight Toolkit for Bounding Boxes

With no prior knowledge of machine learning or device-specific deployment, you can deploy a computer vision model to a range of devices and environments using Roboflow Inference.

With no prior knowledge of machine learning or device-specific deployment, you can deploy a computer vision model to a range of devices and environments using Roboflow Inference.

Packaged version of the Yolov5 object detector

With no prior knowledge of machine learning or device-specific deployment, you can deploy a computer vision model to a range of devices and environments using Roboflow Inference CLI.

Pymba is a Python wrapper for Allied Vision's Vimba C API.

With no prior knowledge of machine learning or device-specific deployment, you can deploy a computer vision model to a range of devices and environments using Roboflow Inference.

Ultralytics HUB Client SDK.

Image Acquisition Library for GenICam-based Machine Vision System

Catalyst. Accelerated deep learning R&D with PyTorch.

A toolkit for making real world machine learning and data analysis applications

A research and production integrated edge-cloud library for federated/distributed machine learning at anywhere at any scale.

The realtime communication library for Python

TorchSeg: Semantic Segmentation models for PyTorch

Easily turn a set of image urls to an image dataset

BoxMOT: pluggable SOTA tracking modules for segmentation, object detection and pose estimation models

A lightweight Computer Vision library for high-performance AI research - Modern Computer Vision on the Fly.

HuggingFace utilities for Ultralytics/YOLOv8.

Computer vision models for PyTorch

A rastervision plugin that adds geospatial machine learning pipelines

An inference runtime offering GPU-class performance on CPUs and APIs to integrate ML into your application

Easily computing clip embeddings and building a clip retrieval system with them

This is the module for detecting and classifying text on rama pictures

Defining the Future of 3D Machine Vision

Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes

Find issues in image datasets

With no prior knowledge of machine learning or device-specific deployment, you can deploy a computer vision model to a range of devices and environments using Roboflow Inference.

🔥LIT: The Learning Interpretability Tool

With no prior knowledge of machine learning or device-specific deployment, you can deploy a computer vision model to a range of devices and environments using Roboflow Inference.

AI Data Infrastructure: Declarative, Multimodal, and Incremental

Torchreid-Pip: Deep learning person re-identification in PyTorch.

Extracts the Machine Readable Zone (MRZ) data from document images

Deploy inference models served for Artificial Intelligence solutions including but not limited to Pixano.

Packaged version of the ByteTrack repository

Declarative machine learning: End-to-end machine learning pipelines using data-driven configurations.

:warning: The `yolov8` package is a placeholder, not the official Ultralytics version. Please install the official `ultralytics` package via `pip install ultralytics` instead.

Viewing and rendering of sequences of 3D data.

doc2text drastically improves the extraction of text from images by fixing resolution, text area (crop), and skew.

An inference runtime offering GPU-class performance on CPUs and APIs to integrate ML into your application

A package designed to predict static pose and detect falls with 2D RGB Camera in well lit indoor environments.

Agnostic Computer Vision Framework

Packaged version of the Strongsort repository