Computer vision libraries

72 libraries, 30 updated this year

Libraries that read images and video. OpenCV is the foundation almost everything else builds on, with bindings for C++, Python, Java, .NET (Emgu CV), Go (GoCV) and MATLAB (mexopencv). For deep learning models, Ultralytics YOLO handles real-time object detection, detectron2 covers detection and segmentation research, and timm collects hundreds of pretrained image backbones.

Face analysis has its own cluster: face_recognition and dlib for detection and embeddings, deepface for attribute analysis, and OpenFace. For training data, albumentations augments images and segmentation_models.pytorch packages common segmentation architectures with pretrained encoders.

Some entries are classic signal processing toolboxes (curvelets, shearlets) or early research code that predates deep learning; check the activity column before you depend on one.

72 libraries

OpenCVopencv/opencv

OpenCV has C++, C, Python, Java and MATLAB interfaces and supports Windows, Linux, Android and Mac OS.

C++GeneralDeep learning
91.3k updated today

YOLOv8ultralytics/ultralytics

Ultralytics' YOLOv8 implementation with C++ support for real-time object detection and tracking, optimized for edge devices.

CDeep learning
62.8k updated today

face_recognitionageitgey/face_recognition

Face recognition library that recognizes and manipulates faces from Python or from the command line.

Python
56.9k updated 4mo ago

timmrwightman/pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more.

PythonDeep learning
37.2k updated 4d ago

detectron2facebookresearch/detectron2

FAIR's next-generation research platform for object detection and segmentation. It is a ground-up rewrite of the previous version, Detectron, and is powered by the PyTorch deep learning framework.

PythonDeep learning
34.9k updated 4d ago

OpenposeCMU-Perceptual-Computing-Lab/openpose

A real-time multi-person keypoint detection library for body, face, hands, and foot estimation

34.5k updated 2y ago

MLXml-explore/mlx

MLX is an array framework for machine learning on Apple silicon, developed by Apple machine learning research.

Python
28.9k updated 1d ago

Detectronfacebookresearch/Detectron Archived

FAIR's software system that implements state-of-the-art object detection algorithms, including Mask R-CNN. It is written in Python and powered by the Caffe2 deep learning framework.

PythonDeep learning
26.3k archived

deepfaceserengil/deepface

A lightweight face recognition and facial attribute analysis (age, gender, emotion and race) framework for Python covering cutting-edge models such as VGG-Face, FaceNet, OpenFace, DeepFace, DeepID, Dlib and ArcFace.

Python
23.6k updated today
O

OpenFacecmusatyalab/openface

Free and open source face recognition with deep neural networks.

PythonDeep learning
15.4k updated 10d ago

albumentationsalbu/albumentations Archived

А fast and framework agnostic image augmentation library that implements a diverse set of augmentation techniques. Supports classification, segmentation, detection out of the box. Was used to win a number of Deep Learning competitions at Kaggle, Topcoder and those that were a part of the CVPR workshops.

PythonDeep learning
15.3k archived

DLibdavisking/dlib

DLib has C++ and Python interfaces for face detection and training general object detectors.

C++GeneralDeep learningData & viz
14.5k updated 3d ago

segmentation_models.pytorchqubvel/segmentation_models.pytorch

A PyTorch-based toolkit that offers pre-trained segmentation models for computer vision tasks. It simplifies the development of image segmentation applications by providing a collection of popular architecture implementations, such as UNet and PSPNet, along with pre-trained weights, making it easier for researchers and developers to achieve high-quality pixel-level object segmentation in images.

PythonDeep learning
11.8k updated 1d ago

Turi Createapple/turicreate Archived

Machine learning from Apple. Turi Create simplifies the development of custom machine learning models. You don't have to be a machine learning expert to add recommendations, object detection, image classification, image similarity or activity classification to your app.

PythonGeneralDeep learning
11.2k archived

AutoGluonawslabs/autogluon

AutoML for Image, Text, Tabular, Time-Series, and MultiModal Data.

PythonGeneralDeep learningNLPMLOps
10.8k updated 2d ago

Hubactiveloopai/Hub

Fastest unstructured dataset management for TensorFlow/PyTorch. Stream & version-control data. Store even petabyte-scale data in a single numpy-like array on the cloud accessible on any machine. Visit activeloop.ai for more info.

PythonGeneralDeep learningLLMsMLOps
9.3k updated 5mo ago

Exadel CompreFaceexadel-inc/CompreFace

Face recognition system that can be easily integrated into any system without prior machine learning skills. CompreFace provides REST API for face recognition, face verification, face detection, face mask detection, landmark detection, age, and gender recognition and is easily deployed with docker.

Python
8.4k updated 2y ago

GoCVhybridgroup/gocv

Package for computer vision using OpenCV 4 and beyond.

Go
7.5k updated 4mo ago

CCVliuliu/ccv

C-based/Cached/Core Computer Vision Library, A Modern Computer Vision Library.

C
7.2k updated 1d ago

Serpent.AISerpentAI/SerpentAI Archived

Serpent.AI is a game agent framework that allows you to turn any video game you own into a sandbox to develop AI and machine learning experiments. For both researchers and hobbyists.

PythonDeep learningRL
7k archived

Scikit-Optguofei9987/scikit-opt

Swarm Intelligence in Python (Genetic Algorithm, Particle Swarm Optimization, Simulated Annealing, Ant Colony Algorithm, Immune Algorithm, Artificial Fish Swarm Algorithm in Python)

Python
6.7k updated 7mo ago

Scikit-Imagescikit-image/scikit-image

A collection of algorithms for image processing in Python.

Python
6.6k updated 1d ago

pytessarctmadmaze/pytesseract

Python-tesseract is an optical character recognition (OCR) tool for python. That is, it will recognize and "read" the text embedded in images. Python-tesseract is a wrapper for Google's Tesseract-OCR Engine.

Python
6.4k updated 5d ago

Swift AISwift-AI/Swift-AI

Highly optimized artificial intelligence and machine learning library written in Swift.

SwiftGeneralDeep learning
6k updated 9y ago

segmentation_modelsqubvel/segmentation_models

A TensorFlow Keras-based toolkit that offers pre-trained segmentation models for computer vision tasks. It simplifies the development of image segmentation applications by providing a collection of popular architecture implementations, such as UNet and PSPNet, along with pre-trained weights, making it easier for researchers and developers to achieve high-quality pixel-level object segmentation in images.

PythonDeep learning
4.9k updated 2y ago

imutilsjrosebr1/imutils

A library containing Convenience functions to make basic image processing operations such as translation, rotation, resizing, skeletonization, and displaying Matplotlib images easier with OpenCV and Python.

Python
4.6k updated 2y ago

Accord.NETaccord-net/framework Archived

Together with AForge.NET, this library can provide image processing and computer vision algorithms to Windows, Windows RT and Windows Phone. Some components are also available for Java and Android.

.NETGeneralDeep learning
4.5k archived

Deep High-Resolution-Netleoxiaobin/deep-high-resolution-net.pytorch

A PyTorch implementation of CVPR2019 paper "Deep High-Resolution Representation Learning for Human Pose Estimation"

PythonDeep learning
4.5k updated 2y ago

lightlylightly-ai/lightly

Lightly is a computer vision framework for self-supervised learning.

PythonDeep learning
3.8k updated 1d ago

Towheetowhee-io/towhee

A Python module that encode unstructured data into embeddings.

PythonGeneral
3.4k updated 2y ago

Catalystcatalyst-team/catalyst

High-level utils for PyTorch DL & RL research. It was developed with a focus on reproducibility, fast experimentation and code/ideas reusing. Being able to research/develop something new, rather than write another regular train loop.

PythonGeneralDeep learningRL
3.4k updated 3mo ago

computer-vision-in-actionCharmve/computer-vision-in-action

As known as L0CV, is a new generation of computer vision open source online learning media, a cross-platform interactive learning framework integrating graphics, source code and HTML. the L0CV ecosystem — Notebook, Datasets, Source Code, and from Diving-in to Advanced — as well as the L0CV Hub.

PythonDeep learning
2.9k updated 2y ago
S

SimpleCVsightmachine/SimpleCV

An open source computer vision framework that gives access to several high-powered computer vision libraries, such as OpenCV. Written on Python and runs on Mac, Windows, and Ubuntu Linux.

Python
2.7k updated 1y ago

DeepDetectjolibrain/deepdetect

A machine learning API and server written in C++11. It makes state of the art machine learning easy to work with and integrate into existing applications.

C++GeneralDeep learning
2.6k updated today
E

Emgu CVemgucv/emgucv

Cross platform wrapper of OpenCV which can be compiled in Mono to be run on Windows, Linus, Mac OS X, iOS, and Android.

.NET
2.3k updated 2d ago

retinafaceserengil/retinaface

Deep learning based cutting-edge facial detector for Python coming with facial landmarks

PythonDeep learning
2.1k updated 15d ago

CosmoEdgecosmo-wander-ai/cosmo-edge

A C++ edge video AI engine for RTSP ingestion, CV/VLM inference, visual pipeline orchestration, alarms, and event delivery on Sophon and Rockchip NPUs.

C++
2k updated 1d ago

PCVjesolem/PCV Deprecated

Open source Python module for computer vision.

Python
2k updated 5y ago

PyTorch Lightning BoltsPyTorchLightning/pytorch-lightning-bolts Archived

Toolbox of models, callbacks, and datasets for AI/ML researchers.

PythonGeneralDeep learningNLP
1.8k archived

LightlyTrainlightly-ai/lightly-train

Pretrain computer vision models on unlabeled data for industrial applications

PythonDeep learning
1.7k updated today

OpenMetricLearningOML-Team/open-metric-learning

A PyTorch-based framework to train and validate the models producing high-quality embeddings.

PythonGeneralDeep learning
998 updated 11mo ago

TF-GANtensorflow/gan Archived

TF-GAN is a lightweight library for training and evaluating Generative Adversarial Networks (GANs).

Python
961 archived

grtnickgillian/grt

The Gesture Recognition Toolkit (GRT) is a cross-platform, open-source, C++ machine learning library designed for real-time gesture recognition.

C++
894 updated 6y ago

neural-style-ptProGamerGov/neural-style-pt

A PyTorch implementation of Justin Johnson's neural-style (neural style transfer).

PythonDeep learning
855 updated 4y ago

Lucentgreentfrapp/lucent

Tensorflow and OpenAI Clarity's Lucid adapted for PyTorch.

Python
668 updated 1y ago
m

mexopencvkyamagu/mexopencv Archived

Collection and a development kit of MATLAB mex functions for OpenCV library.

MATLAB
660 archived

Detectoalankbi/detecto

Train and run a computer vision model with 5-10 lines of code.

PythonDeep learning
623 updated 2y ago

VIGRAukoethe/vigra

VIGRA is a genertic cross-platform C++ computer vision and machine learning library for volumes of arbitrary dimensionality with Python bindings.

443 updated 6mo ago

joliGENjolibrain/joliGEN

Generative AI Image Toolset with GANs and Diffusion for Real-World Applications.

PythonDeep learning
290 updated 8d ago

Backpropbackprop-ai/backprop

Backprop makes it simple to use, finetune, and deploy state-of-the-art ML models.

PythonGeneralNLP
240 updated 5y ago
Esc

Type to search every entry. ↑↓ to move, Enter to open.