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.