An open API service indexing awesome lists of open source software.

awesome-production-machine-learning

A curated list of awesome open source libraries to deploy, monitor, version and scale your machine learning
https://github.com/EthicalML/awesome-production-machine-learning

Last synced: 14 days ago
JSON representation

Filtered by computer-vision. Clear filters

AutoML

  • AutoGluon - Automated feature, model, and hyperparameter selection for tabular, image, and text data on top of popular machine learning libraries (Scikit-Learn, LightGBM, CatBoost, PyTorch, MXNet).

Data Annotation and Synthesis

  • COCO Annotator - annotator.svg?style=social) - Web-based image segmentation tool for object detection, localization and keypoints
  • CVAT - ai/cvat.svg?style=social) - CVAT (Computer Vision Annotation Tool) is OpenCV's web-based annotation tool for both videos and images for computer algorithms.
  • Label Studio - studio.svg?style=social) - Multi-domain data labeling and annotation tool with standardized output format.

Data Labelling and Synthesis

  • makesense.ai - sense.svg?style=social) - Free to use online tool for labelling photos. Prepared labels can be downloaded in one of multiple supported formats.
  • PixelAnnotationTool - Image annotation tool with ability to "colour" on the images to select labels for segmentation. Process is semi-automated with the watershed marked algorithm of OpenCV

Data Pipeline

  • Towhee - io/towhee.svg?style=social) - General-purpose machine learning pipeline for generating embedding vectors using one or many ML models.
  • Pixeltable - source Python library providing declarative, incremental data infrastructure for building and managing multimodal AI workloads.

Deployment and Serving

  • Inference - A fast, production-ready inference server for computer vision supporting deployment of many popular model architectures and fine-tuned models. With Inference, you can deploy models such as YOLOv5, YOLOv8, CLIP, SAM, and CogVLM on your own hardware using Docker.
  • OpenVINO - OpenVINO is an open-source toolkit for optimizing and deploying AI inference.
  • DeepSparse - DeepSparse is a sparsity-aware deep learning inference runtime for CPUs.
  • CosmoEdge - wander-ai/cosmo-edge.svg?cacheSeconds=172800) - A C++ edge video AI engine for production deployments, combining RTSP ingestion, CV/VLM inference, visual orchestration, alarms, and event delivery across Sophon and Rockchip NPUs.

Evaluation and Monitoring

  • VLMEvalKit - compass/VLMEvalKit.svg?style=social) - VLMEvalKit is an open-source evaluation toolkit of large vision-language models (LVLMs).
  • SimplerEnv - env/SimplerEnv.svg?style=social) - SimplerEnv is a simulated manipulation policy evaluation environments for real robot setups.

Explainability and Fairness

  • Lightly - ai/lightly.svg?style=social) - A python framework for self-supervised learning on images. The learned representations can be used to analyze the distribution in unlabeled data and rebalance datasets.

Industry Strength Computer Vision

  • Deep Lake - Deep Lake is a data infrastructure optimized for computer vision.
  • supervision - Supervision is a Python library designed for efficient computer vision pipeline management, providing tools for annotation, visualization, and monitoring of models.
  • LightlyTrain - ai/lightly-train.svg?style=social) - Pretrain computer vision models on unlabeled data for industrial applications.
  • Kornia - Kornia is a differentiable computer vision library built on PyTorch that provides a rich set of differentiable image processing and geometric vision algorithms.
  • CameraTraps - CameraTraps (PyTorch Wildlife) is a collaborative deep learning framework for wildlife image analysis, providing detection and classification models trained on large-scale camera trap datasets.

Industry Strength Robotics

  • AI2-THOR - AI2-THOR is a near photo-realistic interactable framework for AI agents.
  • Habitat-Sim - sim.svg?cacheSeconds=86400) - Habitat-Sim is a flexible, high-performance 3D simulator for Embodied AI research.

Industry Strength Visualisation

  • Spotlight - Spotlight helps you to identify critical data segments and model failure modes. It enables you to build and maintain reliable machine learning models by curating high-quality datasets.
  • Rerun - io/rerun.svg?cacheSeconds=86400) - Rerun is an open-source SDK for logging, storing, querying, and visualizing multimodal data, designed for robotics, computer vision, and spatial AI.

Model, Data and Experiment Tracking

  • Catalyst - team/catalyst.svg?style=social) - High-level utils for PyTorch DL & RL research. It was developed with a focus on reproducibility, fast experimentation and code/ideas reusing.

Model Training and Orchestration

  • Ludwig - ai/ludwig.svg?style=social) - Ludwig is a low-code framework for building custom AI models like LLMs and other deep neural networks.

Optimized Computation

  • OpenFlamingo - OpenFlamingo is an open-source framework for training large multimodal models.