Open-source computer-vision software
2359 open-source libraries, tools and applications tagged computer-vision, ranked by how closely computer-vision is their primary purpose, then by adoption and maintenance health. Each entry links to a page with sources, use cases and when to choose or avoid it.
2359 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| cmdbug/YOLOv5_NCNN A mobile demo application that deploys the ncnn inference framework on Android and iOS, running a variety of computer vision models includi… | 32 | 1572 | maintenance |
| microsoft/SoM A research toolbox from Microsoft implementing Set-of-Mark (SoM) visual prompting, which overlays numbered spatial marks on images to impro… | 18 | 1572 | maintenance |
| IDEA-Research/MaskDINO Official PyTorch implementation of Mask DINO, a unified transformer-based framework for object detection and segmentation, built on detectr… | 23 | 1571 | maintenance |
| NVlabs/noise2noise Official TensorFlow implementation of the Noise2Noise ICML 2018 paper, which trains image restoration networks using only corrupted (noisy)… | 32 | 1568 | maintenance |
| HumanAIGC/EMO EMO (Emote Portrait Alive) is a research codebase from Alibaba's Institute for Intelligent Computing that generates expressive talking port… | 26 | 7580 | experimental |
| AlexHex7/Non-local_pytorch A PyTorch implementation of the Non-local Neural Block from the paper 'Non-local Neural Networks', providing multiple variants (concatenati… | 32 | 1566 | maintenance |
| thu-ml/prolificdreamer Official PyTorch implementation of ProlificDreamer, a NeurIPS 2023 method for high-fidelity text-to-3D generation using Variational Score D… | 30 | 1564 | maintenance |
| QingyongHu/RandLA-Net Official TensorFlow implementation of RandLA-Net, a neural architecture for efficient semantic segmentation of large-scale 3D point clouds,… | 32 | 1563 | maintenance |
| V2AI/Det3D Det3D is a PyTorch-based toolbox for 3D object detection from point clouds, offering implementations of models like PointPillars, SECOND, a… | 32 | 1562 | maintenance |
| facebookresearch/DeepSDF DeepSDF is Facebook Research's official implementation of the CVPR 2019 paper on learning continuous signed distance functions for 3D shape… | 10 | 1562 | maintenance |
| genforce/interfacegan InterFaceGAN is a Python research library that interprets the latent space of pretrained GANs (PGGAN, StyleGAN) to find semantic subspaces … | 32 | 1561 | maintenance |
| msracver/FCIS FCIS is the official MXNet implementation of the CVPR 2017 paper 'Fully Convolutional Instance-aware Semantic Segmentation', which won firs… | 32 | 1560 | maintenance |
| vt-vl-lab/FGVC FGVC is a PyTorch implementation of the ECCV 2020 paper 'Flow-edge Guided Video Completion'. It completes missing regions in videos by extr… | 32 | 1554 | maintenance |
| JingyunLiang/VRT VRT is the official PyTorch implementation of the paper 'VRT: A Video Restoration Transformer', a transformer-based model for video restora… | 65 | 1553 | maintenance |
| DevashishPrasad/CascadeTabNet CascadeTabNet is a PyTorch/mmdetection implementation of a CVPR 2020 paper for end-to-end table detection and structure recognition from im… | 32 | 1551 | maintenance |
| JiaRenChang/PSMNet PSMNet is the official PyTorch implementation of the CVPR 2018 paper 'Pyramid Stereo Matching Network' for stereo depth estimation. It uses… | 32 | 1551 | maintenance |
| pytorch/QNNPACK QNNPACK is a mobile-optimized C library of high-performance kernels for 8-bit quantized neural network operators such as convolution, pooli… | 10 | 1551 | maintenance |
| Javacr/PyQt5-YOLOv5 A desktop GUI application built with PyQt5 that wraps YOLOv5 (v6.1) object detection models. It supports running detection on images, video… | 32 | 1548 | maintenance |
| YoYo000/MVSNet MVSNet is a deep learning architecture for depth map inference from unstructured multi-view images, and R-MVSNet is its recurrent extension… | 32 | 1548 | maintenance |
| google-research/big_transfer Official repository for the Big Transfer (BiT) paper, providing ResNet models pre-trained on ImageNet and ImageNet-21k for transfer learnin… | 10 | 1541 | maintenance |
| cruxopen/openISP An open-source Python implementation of an Image Signal Processor (ISP) pipeline that converts RAW sensor images to RGB/YUV. It simulates h… | 32 | 1537 | maintenance |
| dandelin/ViLT Official PyTorch code for the ICML 2021 paper ViLT, a vision-and-language transformer that performs multimodal pre-training without convolu… | 23 | 1537 | maintenance |
| gpleiss/efficient_densenet_pytorch A memory-efficient PyTorch implementation of DenseNets that uses gradient checkpointing to reduce feature map memory consumption from quadr… | 32 | 1536 | maintenance |
| lucidrains/lambda-networks A PyTorch implementation of Lambda Networks, featuring the λ layer that models long-range interactions by transforming contexts into linear… | 23 | 1530 | maintenance |
| mzucker/page_dewarp A Python command-line tool that dewarps photos of curled or warped book pages using a cubic sheet optimization model, flattening them into … | 32 | 1529 | maintenance |
| yu4u/age-gender-estimation A Keras/TensorFlow implementation of a convolutional neural network that estimates age and gender from face images, trained on the IMDB-WIK… | 23 | 1521 | maintenance |
| zhan-xu/RigNet RigNet is a PyTorch implementation of the SIGGRAPH 2020 paper on neural rigging for articulated 3D characters. It takes a character mesh as… | 32 | 1520 | maintenance |
| junyanz/BicycleGAN A PyTorch implementation of BicycleGAN, a model for multimodal image-to-image translation that generates diverse outputs from a single inpu… | 32 | 1516 | maintenance |
| tanluren/yolov3-channel-and-layer-pruning A Python toolkit built on ultralytics/yolov3 that implements channel pruning, layer pruning, and knowledge distillation for YOLOv3/v4 (incl… | 32 | 1514 | maintenance |
| krasserm/super-resolution A TensorFlow 2.x implementation of EDSR, WDSR, and SRGAN models for single image super-resolution, with a high-level training API and DIV2K… | 32 | 1514 | maintenance |
| Eric-mingjie/rethinking-network-pruning A PyTorch research codebase reproducing the ICLR 2019 paper 'Rethinking the Value of Network Pruning', which shows pruned models trained fr… | 32 | 1513 | maintenance |
| ShuffleNet Series A collection of ShuffleNet-series efficient convolutional neural network models (V1, V2, V2+, Large, ExLarge) plus NAS-derived backbones li… | 32 | 1511 | maintenance |
| open-mmlab/Multimodal-GPT Multimodal-GPT is an open-source project for training a multimodal chatbot that combines vision and language instructions, built on OpenFla… | 30 | 1511 | maintenance |
| yulunzhang/RCAN PyTorch implementation of RCAN, a very deep residual channel attention network for single image super-resolution from an ECCV 2018 paper. I… | 35 | 1510 | maintenance |
| compphoto/BoostingMonocularDepth A Python research implementation for boosting monocular depth estimation to high resolution using a double-estimation merging operator, sup… | 32 | 1508 | maintenance |
| SOLIDER SOLIDER is a semantic-controllable self-supervised learning framework that learns general human representations from massive unlabeled huma… | 32 | 1508 | maintenance |
| luuuyi/CBAM.PyTorch A non-official PyTorch re-implementation of the CBAM (Convolutional Block Attention Module) paper from ECCV 2018. It provides channel and s… | 32 | 1504 | maintenance |
| AITTSMD/MTCNN-Tensorflow A TensorFlow reproduction of MTCNN (Multi-task Cascaded Convolutional Networks) for joint face detection and facial landmark alignment. It … | 32 | 1502 | maintenance |
| vacancy/Synchronized-BatchNorm-PyTorch A PyTorch library implementing Synchronized Batch Normalization, which computes batch statistics across all GPUs during multi-device traini… | 32 | 1501 | maintenance |
| CNN Image Retrieval A PyTorch toolbox for training and evaluating convolutional neural networks for image retrieval, implementing the authors' TPAMI 2018 and E… | 23 | 1500 | maintenance |
| AIRLegend/aitrack AITrack is a free 6DoF head tracking application that uses a webcam and neural networks to estimate head position and rotation. It streams … | 23 | 1496 | maintenance |
| DeepVoltaire/AutoAugment An unofficial Python implementation of the AutoAugment data augmentation policies learned for ImageNet, CIFAR-10, and SVHN, built on Pillow… | 32 | 1493 | maintenance |
| cvg/pixel-perfect-sfm pixsfm is a Python package with a C++ core that improves Structure-from-Motion and visual localization accuracy by refining keypoints, came… | 23 | 1492 | maintenance |
| chengdazhi/Deformable-Convolution-V2-PyTorch A PyTorch implementation of Deformable Convolution V2 (DCNv2) custom CUDA operators, ported from the original MXNet implementation. It prov… | 32 | 1486 | maintenance |
| speedinghzl/CCNet Official PyTorch implementation of CCNet, a Criss-Cross Attention network for semantic segmentation published at ICCV 2019 and TPAMI 2020. … | 32 | 1486 | maintenance |
| ageitgey/show-facebook-computer-vision-tags A simple Chrome extension that overlays the automated computer vision tags Facebook generates for images directly on your Facebook timeline… | 32 | 1485 | maintenance |
| Sharpiless/Yolov5-deepsort-inference A Python library combining YOLOv5 object detection with DeepSort multi-object tracking to detect, track, and count vehicles and pedestrians… | 62 | 1483 | maintenance |
| chainer/chainercv ChainerCV is a Python library providing tools to train and run neural networks for computer vision tasks on top of the Chainer framework. I… | 10 | 1481 | maintenance |
| shouzhong/Scanner An Android scanning library that recognizes QR codes, barcodes, ID cards, bank cards, license plates, text, driving licenses, and NSFW imag… | 23 | 1478 | maintenance |
| Image Captioning A PyTorch research codebase for image captioning, supporting self-critical sequence training, bottom-up features, transformer captioning mo… | 32 | 1477 | maintenance |
| faustomorales/keras-ocr A Python library packaging the CRAFT text detector and a Keras CRNN text recognition model into a high-level OCR pipeline. It supports pret… | 39 | 1473 | maintenance |
| lxztju/pytorch_classification A complete PyTorch image classification codebase built on torchvision, covering training, prediction, TTA, model ensembling, knowledge dist… | 32 | 1470 | maintenance |
| Up-Down Image Captioning System A bottom-up attention model based on Faster R-CNN with ResNet-101 trained on Visual Genome, producing features for salient image regions. T… | 32 | 1469 | maintenance |
| CSAILVision/LabelMeAnnotationTool The source code for LabelMe, a web-based image annotation tool written in JavaScript that runs on your own Apache server. It lets users dra… | 32 | 1469 | maintenance |
| mmatl/pyrender Pyrender is a pure Python, glTF 2.0-compliant OpenGL renderer for physically-based rendering and visualization of 3D scenes. It includes an… | 32 | 1469 | maintenance |
| szagoruyko/attention-transfer PyTorch reference implementation of the ICLR 2017 paper 'Paying More Attention to Attention', which improves convolutional neural networks … | 32 | 1466 | maintenance |
| PlenOctrees Official PyTorch implementation of pixelNeRF, a CVPR 2021 method that predicts neural radiance fields conditioned on one or few input image… | 32 | 1465 | maintenance |
| HRNet/HigherHRNet-Human-Pose-Estimation Official PyTorch implementation of HigherHRNet, a CVPR 2020 bottom-up multi-person human pose estimation model using scale-aware high-resol… | 32 | 1465 | maintenance |
| Sanster/text_renderer A Python tool that generates synthetic text images with configurable visual effects for training deep learning OCR models like CRNN. It sup… | 32 | 1461 | maintenance |
| datitran/face2face-demo A pix2pix demo that learns from facial landmarks and translates them into a trained target face, with a real-time webcam application. It in… | 32 | 1461 | maintenance |
| una-dinosauria/3d-pose-baseline A TensorFlow implementation of a simple yet effective baseline for 3D human pose estimation from 2D keypoints, published at ICCV 2017. It i… | 32 | 1461 | maintenance |
| buaacyw/GaussianEditor GaussianEditor is a research application for fast, controllable 3D scene editing built on Gaussian Splatting, released as CVPR 2024 code. I… | 27 | 1461 | maintenance |
| wvangansbeke/Unsupervised-Classification PyTorch implementation of SCAN (ECCV 2020), a two-step method for unsupervised image classification that combines self-supervised represent… | 32 | 1460 | maintenance |
| facebookresearch/MaskFormer MaskFormer is a PyTorch/Detectron2-based implementation of the NeurIPS 2021 paper 'Per-Pixel Classification is Not All You Need for Semanti… | 10 | 1459 | maintenance |
| nerfstudio NerfAcc is a PyTorch acceleration toolbox for Neural Radiance Fields (NeRF), focused on efficient sampling in the volumetric rendering pipe… | 23 | 1456 | maintenance |
| abreheret/PixelAnnotationTool A C++ desktop application for quickly annotating images with pixel-wise labels using OpenCV's watershed algorithm. Users draw markers with … | 23 | 1456 | maintenance |
| xuannianz/EfficientDet A Keras/TensorFlow implementation of the EfficientDet object detection model with pretrained COCO and ImageNet weights. It supports trainin… | 32 | 1452 | maintenance |
| hysts/pytorch_image_classification A PyTorch library implementing many image classification architectures (ResNet, DenseNet, WRN, PyramidNet, SENet, etc.) and augmentation te… | 10 | 1451 | maintenance |
| dji-sdk/Tello-Python A collection of Python sample modules for controlling and interacting with the Ryze Tello drone, including command scripting, video streami… | 32 | 1449 | maintenance |
| mit-han-lab/proxylessnas ProxylessNAS is a neural architecture search (NAS) framework that directly searches CNN architectures on the target task and target hardwar… | 32 | 1448 | maintenance |
| megvii-research/ML-GCN A PyTorch implementation of ML-GCN, the CVPR 2019 paper 'Multi-Label Image Recognition with Graph Convolutional Networks'. It provides trai… | 32 | 1447 | maintenance |
| BachiLi/redner redner is a differentiable Monte Carlo ray tracer that computes exact gradients of rendered images with respect to arbitrary scene paramete… | 32 | 1446 | maintenance |
| ZPdesu/Barbershop Official PyTorch implementation of Barbershop (SIGGRAPH Asia 2021), a GAN-inversion-based method for compositing images using segmentation … | 32 | 1440 | maintenance |
| mchong6/JoJoGAN JoJoGAN is the official PyTorch implementation of a one-shot face stylization method that finetunes a pretrained StyleGAN using GAN inversi… | 32 | 1440 | maintenance |
| toandaominh1997/EfficientDet.Pytorch A PyTorch implementation of EfficientDet, a scalable and efficient object detection model from a 2019 Google Research paper. It supports Ef… | 10 | 1439 | maintenance |
| VDIGPKU/M2Det M2Det is a PyTorch implementation of a single-shot object detector based on a Multi-Level Feature Pyramid Network (MLFPN), published at AAA… | 32 | 1438 | maintenance |
| xuebinqin/BASNet BASNet is the official PyTorch implementation of the CVPR 2019 paper 'BASNet: Boundary-Aware Salient Object Detection', a deep learning mod… | 32 | 1436 | maintenance |
| wenhaochai/StableVideo StableVideo is the official ICCV 2023 implementation of text-driven, consistency-aware diffusion-based video editing built on ControlNet an… | 31 | 1436 | maintenance |
| yangyanli/PointCNN PointCNN is a deep learning framework for feature learning from 3D point clouds, applying convolution on X-transformed points to handle the… | 51 | 1434 | maintenance |
| hollance/CoreMLHelpers A collection of Swift types and helper functions that simplify working with Apple's Core ML framework, such as image-to-CVPixelBuffer conve… | 32 | 1434 | maintenance |
| MSPaintIDE/MSPaintIDE MS Paint IDE is a novelty-but-functional IDE that uses a custom OCR engine to read code written in MS Paint images, then highlights, parses… | 23 | 1433 | maintenance |
| TSDF Fusion A lightweight Python script that fuses multiple registered RGB-D images into a projective TSDF voxel volume, from which high-quality 3D sur… | 32 | 1432 | maintenance |
| KaiyangZhou/Dassl.pytorch Dassl is a PyTorch toolbox for research in domain adaptation, domain generalization, and semi-supervised learning. It provides modular comp… | 32 | 1431 | maintenance |
| vlfeat/matconvnet MatConvNet is a MATLAB toolbox implementing convolutional neural networks (CNNs) for computer vision applications. It supports training and… | 32 | 1430 | maintenance |
| qfgaohao/pytorch-ssd A PyTorch implementation of the SSD (Single Shot MultiBox Detector) object detection algorithm with MobileNetV1, MobileNetV2, and VGG backb… | 32 | 1430 | maintenance |
| nianticlabs/simplerecon SimpleRecon is the reference PyTorch implementation of an ECCV 2022 paper for multi-view stereo depth estimation and 3D reconstruction that… | 38 | 1429 | maintenance |
| theAIGuysCode/yolov4-deepsort A Python implementation of multi-object tracking that combines YOLOv4 object detection with the Deep SORT tracking algorithm using TensorFl… | 32 | 1427 | maintenance |
| open-mmlab/mmhuman3d MMHuman3D is an open-source PyTorch-based toolbox and benchmark for 3D human parametric models (e.g., SMPL, SMPL-X) in computer vision and … | 23 | 1426 | maintenance |
| rmokady/CLIP_prefix_caption Official implementation of ClipCap, a CLIP-based image captioning model that maps CLIP image encodings to a GPT-2 prefix to generate captio… | 32 | 1423 | maintenance |
| torchgan/torchgan TorchGAN is a PyTorch-based research framework for designing and training Generative Adversarial Networks. It provides modular building blo… | 23 | 1422 | maintenance |
| sfzhang15/RefineDet RefineDet is a C++/Caffe implementation of the CVPR 2018 single-shot object detection model that refines anchors to combine one-stage speed… | 32 | 1421 | maintenance |
| VerticalResearchGroup/miaow MIAOW is an open source GPU implementation of the AMD Southern Islands ISA written in Verilog. It was developed as a research project at th… | 36 | 1420 | maintenance |
| BangguWu/ECANet Official PyTorch implementation of ECA-Net, an efficient channel attention module for deep convolutional neural networks published at CVPR … | 32 | 1418 | maintenance |
| GOATmessi8/RFBNet PyTorch implementation of RFBNet, a Receptive Field Block Net object detector presented at ECCV 2018. It enhances SSD-style detectors with … | 32 | 1416 | maintenance |
| cdpierse/transformers-interpret A Python library providing model explainability for Hugging Face Transformers models, built on Captum. It explains text classification, que… | 23 | 1416 | maintenance |
| chrischoy/3D-R2N2 3D-R2N2 is a PyTorch-based implementation of a recurrent neural network that reconstructs voxelized 3D models of objects from one or multip… | 32 | 1410 | maintenance |
| bermanmaxim/LovaszSoftmax Standalone PyTorch and TensorFlow implementations of the Lovász-Softmax and Lovász Hinge loss functions for optimizing the Jaccard index (I… | 32 | 1409 | maintenance |
| yu-changqian/TorchSeg A fast, modular PyTorch reference implementation for training and evaluating semantic segmentation models such as FCN, DFN, BiSeNet, PSPNet… | 23 | 1408 | maintenance |
| moeiscool/Shinobi Shinobi CE is a free, open-source CCTV/NVR platform written in Node.js for recording and managing IP security cameras. It supports RTSP/ONV… | 23 | 1408 | maintenance |
| amazon-science/patchcore-inspection Official implementation of PatchCore, a deep-learning method for industrial image anomaly detection and localization from Roth et al. (2021… | 32 | 1407 | maintenance |