Machine Vision

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

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

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.

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

QUick and DIrty Domain Adaptation

Image augmentation library for deep neural networks

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

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.

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 CLI.

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.

A rastervision plugin that adds geospatial machine learning pipelines

Ultralytics HUB Client SDK.

Image Acquisition Library for GenICam-based Machine Vision System

The realtime communication library for Python

RF100-VL Dataset Interface

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

RF-DETR

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

Catalyst. Accelerated deep learning R&D with PyTorch.

TorchSeg: Semantic Segmentation models for PyTorch

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

HuggingFace utilities for Ultralytics/YOLOv8.

Computer vision models for PyTorch

Easily turn a set of image urls to an image dataset

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

Automatically built Ultralytics package with python-opencv-headless dependency instead of python-opencv

Agnostic Computer Vision Framework

Defining the Future of 3D Machine Vision

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

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

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

YOLO Dataset Debugger (yololint) is a tool for automatic validation and diagnostics of YOLO-format datasets. It helps you quickly detect common errors, inconsistencies, and missing files in your dataset structure and annotations before you start model training. With clear reports and easy usage, you save time and ensure your dataset is ready for deep learning projects.

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

🔥LIT: The Learning Interpretability Tool

Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models

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.

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

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.

Find issues in image datasets

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

AI Data Infrastructure: Declarative, Multimodal, and Incremental

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

Video streaming inference framework, integrating image algorithms and models for real-time/offline video structuring

Image dataset analyzer using image embedding models and clustering methods.

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

A unified library for object tracking featuring clean room re-implementations of leading multi-object tracking algorithms