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 |
|---|---|---|---|
| facebookresearch/TimeSformer Official PyTorch implementation of TimeSformer, a space-time attention transformer model for video understanding from an ICML 2021 paper. I… | 10 | 1862 | maintenance |
| StackGAN TensorFlow implementation of StackGAN, a stacked generative adversarial network that synthesizes photo-realistic images from text descripti… | 32 | 1861 | maintenance |
| magicleap/Atlas Atlas is a PyTorch-based deep learning model from Magic Leap that performs end-to-end 3D scene reconstruction from posed RGB images, produc… | 32 | 1859 | maintenance |
| muesli/smartcrop A pure Go library that finds good image crops for arbitrary crop sizes, based on Jonas Wagner's smartcrop.js. It includes a simple CLI appl… | 23 | 1855 | maintenance |
| WangYueFt/dgcnn Author's reference implementation of Dynamic Graph CNN (DGCNN), a neural network module called EdgeConv for learning on 3D point clouds. It… | 32 | 1848 | maintenance |
| xinshuoweng/AB3DMOT Official Python implementation of AB3DMOT, a simple real-time baseline for 3D multi-object tracking using oriented 3D bounding boxes from L… | 32 | 1846 | maintenance |
| microsoft/Cream A collection of Microsoft research projects on neural architecture search (NAS) and efficient Vision Transformers, including AutoFormer, Ti… | 10 | 1844 | maintenance |
| AlexeyAB/Yolo_mark A Windows and Linux GUI application for drawing bounding boxes around objects in images to create labeled training data for YOLO v2/v3 obje… | 32 | 1842 | maintenance |
| google-deepmind/kinetics-i3d A repository of pre-trained Inflated 3D Convnet (I3D) models for video action classification, trained on the Kinetics dataset, released alo… | 32 | 1839 | maintenance |
| Janspiry/Palette-Image-to-Image-Diffusion-Models An unofficial PyTorch implementation of the Palette paper on image-to-image diffusion models, supporting tasks like inpainting, uncropping,… | 32 | 1833 | maintenance |
| Silent Face Anti-Spoofing An open-source silent face anti-spoofing (liveness detection) project by MiniVision, providing model training code, data preprocessing, tes… | 32 | 1833 | maintenance |
| bubbliiiing/faster-rcnn-pytorch A PyTorch implementation of the Faster R-CNN two-stage object detection model, supporting training on VOC-format datasets with ResNet or VG… | 23 | 1832 | maintenance |
| NVIDIA/semantic-segmentation NVIDIA's PyTorch monorepo implementing the paper 'Hierarchical Multi-Scale Attention for Semantic Segmentation', with pretrained weights an… | 32 | 1831 | maintenance |
| NVlabs/stylegan2-ada NVIDIA's official TensorFlow implementation of StyleGAN2 with adaptive discriminator augmentation (ADA), enabling high-quality GAN training… | 32 | 1829 | maintenance |
| leapmotion/ProjectNorthStar Open-source reference design for Leap Motion's Project North Star augmented reality headset, including mechanical CAD, electrical designs, … | 50 | 1822 | maintenance |
| yassouali/pytorch-segmentation A PyTorch library implementing multiple semantic segmentation models (DeepLab V3+, PSPNet, U-Net, SegNet, FCN, ENet, and others) with datas… | 23 | 1818 | maintenance |
| shunsukesaito/PIFu PyTorch implementation of PIFu (Pixel-Aligned Implicit Function), a deep learning method that reconstructs high-resolution 3D surfaces and … | 32 | 1817 | maintenance |
| sshaoshuai/PointRCNN PyTorch implementation of PointRCNN, a two-stage 3D object detector that generates 3D box proposals directly from raw LiDAR point clouds. I… | 32 | 1817 | maintenance |
| MCG-NJU/VideoMAE Official PyTorch implementation of VideoMAE, a masked autoencoder method for data-efficient self-supervised video pre-training with video t… | 32 | 1803 | maintenance |
| symisc/sod SOD is an embedded, cross-platform computer vision and machine learning library written in C, distributed as a single dependency-free amalg… | 23 | 1802 | maintenance |
| xmartlabs/Bender Bender is a Swift library that provides an abstraction layer over Apple's MetalPerformanceShaders for defining and running neural networks … | 23 | 1800 | maintenance |
| harskish/ganspace GANSpace is a research codebase implementing the NeurIPS 2020 paper on discovering interpretable controls for GANs via PCA in activation sp… | 32 | 1796 | maintenance |
| jkjung-avt/tensorrt_demos A collection of Python demo programs showing how to optimize and run deep learning models (YOLOv4, YOLOv3, SSD, MTCNN, GoogLeNet, MODNet) w… | 46 | 1793 | maintenance |
| dreamoving/dreamoving-project DreaMoving is the official implementation of a diffusion-based controllable video generation framework from Alibaba that produces high-qual… | 27 | 1789 | maintenance |
| ali-vilab/dreamtalk DreamTalk is the official implementation of a diffusion-based framework for generating expressive, audio-driven talking head videos from a … | 27 | 1788 | maintenance |
| ttengwang/Caption-Anything Caption-Anything combines Segment Anything image segmentation, visual captioning, and ChatGPT to generate tailored captions for any object … | 31 | 1777 | maintenance |
| wkentaro/pytorch-fcn A PyTorch implementation of Fully Convolutional Networks (FCN) for semantic segmentation, including training code that reproduces the origi… | 10 | 1777 | maintenance |
| sergiomsilva/alpr-unconstrained An implementation of the ECCV 2018 paper 'License Plate Detection and Recognition in Unconstrained Scenarios', combining a Darknet-based de… | 32 | 1772 | maintenance |
| princeton-vl/CornerNet-Lite CornerNet-Lite is the official PyTorch implementation of the paper 'CornerNet-Lite: Efficient Keypoint Based Object Detection', providing t… | 32 | 1770 | maintenance |
| nywang16/Pixel2Mesh Pixel2Mesh is the official TensorFlow implementation of the ECCV 2018 paper that generates 3D mesh models from a single RGB image using a g… | 32 | 1768 | maintenance |
| CSAILVision/gandissect GANDissect is a PyTorch-based toolkit for visualizing and understanding the internal neurons of generative adversarial networks, showing ho… | 32 | 1765 | maintenance |
| laboshinl/loam_velodyne A ROS implementation of LOAM (Laser Odometry and Mapping), a realtime method for state estimation and 3D mapping using a Velodyne 3D lidar.… | 32 | 1763 | maintenance |
| facebookresearch/votenet VoteNet is the official PyTorch implementation of the ICCV 2019 paper 'Deep Hough Voting for 3D Object Detection in Point Clouds'. It provi… | 10 | 1762 | maintenance |
| WXinlong/SOLO Official PyTorch implementation of SOLO and SOLOv2, box-free fully convolutional methods for instance segmentation published at ECCV 2020 a… | 32 | 1756 | maintenance |
| salesforce/ALBEF Official PyTorch implementation of ALBEF, a vision-and-language pre-training method that aligns image and text representations before fusin… | 10 | 1754 | maintenance |
| Artistic Video Style Transfer A Torch (Lua) implementation of the paper 'Artistic style transfer for videos', which applies the style of a painting to an entire video wh… | 10 | 1753 | maintenance |
| Lightning-Universe/lightning-bolts Lightning Bolts is a toolbox of pre-built models, callbacks, and datasets that extend PyTorch Lightning for AI/ML research and production. … | 10 | 1753 | maintenance |
| mikepound/mazesolving A collection of Python algorithms that solve mazes from input images, created as a companion to a Computerphile video. It supports multiple… | 32 | 1748 | maintenance |
| cszn/DnCNN DnCNN is the official implementation of the TIP 2017 paper 'Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising',… | 32 | 1740 | maintenance |
| DeepCluster DeepCluster is a research codebase from Facebook AI Research implementing the 'Deep Clustering for Unsupervised Learning of Visual Features… | 10 | 1740 | maintenance |
| bearpaw/pytorch-classification A PyTorch library providing a unified interface for training image classification models on CIFAR-10/100 and ImageNet. It includes implemen… | 32 | 1738 | maintenance |
| NVlabs/PWC-Net Official NVIDIA implementation of PWC-Net, a CNN for optical flow estimation using pyramid, warping, and cost volume, released with CVPR 20… | 32 | 1737 | maintenance |
| zijundeng/pytorch-semantic-segmentation A PyTorch library providing implementations of popular semantic segmentation models such as FCN, U-Net, SegNet, PSPNet, GCN, and DUC/HDC. I… | 32 | 1736 | maintenance |
| natethegreate/hent-AI A Python application that automatically detects censor bars and mosaic blurs in illustrated adult content using deep learning (Mask R-CNN) … | 23 | 1734 | maintenance |
| ranahanocka/MeshCNN MeshCNN is a PyTorch library implementing a convolutional neural network that operates directly on 3D triangular meshes, with mesh-specific… | 32 | 1732 | maintenance |
| experiencor/keras-yolo2 A Keras/TensorFlow implementation of the YOLOv2 real-time object detection model with support for training on custom datasets. It offers mu… | 23 | 1731 | maintenance |
| NVIDIA/Cosmos-Tokenizer NVIDIA Cosmos Tokenizer is a suite of neural tokenizers for images and videos that convert visual data into continuous latents or discrete … | 10 | 1729 | maintenance |
| bubbliiiing/unet-pytorch A PyTorch implementation of the U-Net semantic segmentation model with training, prediction, and mIoU evaluation scripts. It supports multi… | 23 | 1726 | maintenance |
| Lightning-Universe/lightning-flash Lightning Flash is a high-level PyTorch library built on PyTorch Lightning that provides ready-made 'recipes' for over 15 AI tasks across 7… | 10 | 1726 | maintenance |
| cvlab-columbia/viper ViperGPT is a research codebase that composes vision-and-language models with code generated by large language models (GPT-3.5/GPT-4) to pe… | 31 | 1720 | maintenance |
| svip-lab/impersonator A PyTorch implementation of Liquid Gating GAN (ICCV 2019) that performs human motion imitation, appearance transfer, and novel view synthes… | 32 | 1718 | maintenance |
| dunbar12138/pix2pix3D pix2pix3D is the official PyTorch implementation of a CVPR 2023 paper on 3D-aware conditional image synthesis. It generates 3D objects (neu… | 31 | 1717 | maintenance |
| PyNET DPED is a Python/TensorFlow implementation of a deep convolutional network approach that translates ordinary smartphone photos into DSLR-qu… | 46 | 1710 | maintenance |
| MaixPy MaixPy-v1 is a MicroPython port for the Kendryte K210 RISC-V AI chip, letting users program Sipeed Maix boards in Python. It provides APIs … | 23 | 1709 | maintenance |
| Fyusion/LLFF LLFF (Local Light Field Fusion) is a TensorFlow implementation of the SIGGRAPH 2019 paper for novel view synthesis from sparse input images… | 32 | 1708 | maintenance |
| koide3/fast_gicp A C++ library of fast GICP-based point cloud registration algorithms, including multi-threaded GICP, voxelized GICP (VGICP), and CUDA-accel… | 36 | 1704 | maintenance |
| i-Code Microsoft's i-Code is a collection of research models and frameworks for integrative, composable multimodal AI spanning vision, language, a… | 10 | 1700 | maintenance |
| hanleyweng/CoreML-in-ARKit A Swift demo/template iOS app that runs CoreML object detection (Inception V3) on live ARKit camera frames and renders 3D text labels above… | 32 | 1695 | maintenance |
| VITA-Group/TransGAN Official PyTorch implementation of TransGAN, a NeurIPS 2021 paper that builds a GAN whose generator and discriminator are both pure transfo… | 32 | 1693 | maintenance |
| jamriska/ebsynth Ebsynth is a fast example-based image synthesis tool that performs style transfer, guided texture synthesis, inpainting, and super-resoluti… | 32 | 1685 | maintenance |
| LoSealL/VideoSuperResolution A Python library (pip-installable as VSR) collecting reimplementation of state-of-the-art single-image and video super-resolution neural ne… | 23 | 1685 | maintenance |
| adobe/antialiased-cnns A PyTorch library providing antialiased CNN models and a BlurPool layer from the ICML 2019 paper 'Making Convolutional Networks Shift-Invar… | 23 | 1684 | maintenance |
| open-mmlab/mmrazor MMRazor is OpenMMLab's model compression toolbox and benchmark built on PyTorch. It provides implementations of neural architecture search,… | 23 | 1683 | maintenance |
| CuriousAI/mean-teacher Reference implementations (TensorFlow and PyTorch) of the Mean Teacher semi-supervised learning method from the NIPS 2017 paper by Tarvaine… | 32 | 1681 | maintenance |
| YuliangXiu/ICON ICON is a PyTorch research implementation of a CVPR 2022 method that reconstructs detailed, animatable 3D clothed human avatars from 2D ima… | 23 | 1681 | maintenance |
| Lam1360/YOLOv3-model-pruning A PyTorch implementation of YOLOv3 channel pruning (network slimming) applied to hand detection on the Oxford Hand dataset. It provides spa… | 32 | 1676 | maintenance |
| argusswift/YOLOv4-pytorch A PyTorch re-implementation of YOLOv4 object detection with variants including attentive YOLOv4 (SEnet, CBAM, CoordAttention) and MobileNet… | 23 | 1674 | maintenance |
| charlesq34/frustum-pointnets Official TensorFlow code release for the CVPR 2018 paper 'Frustum PointNets for 3D Object Detection from RGB-D Data' by Stanford and Nuro r… | 32 | 1672 | maintenance |
| akanazawa/hmr HMR (Human Mesh Recovery) is a TensorFlow implementation of the CVPR 2018 paper 'End-to-end Recovery of Human Shape and Pose', which regres… | 32 | 1671 | maintenance |
| autonomousvision/occupancy_networks Official PyTorch implementation of the CVPR 2019 paper 'Occupancy Networks: Learning 3D Reconstruction in Function Space'. It learns contin… | 32 | 1667 | maintenance |
| natanielruiz/deep-head-pose Hopenet is a PyTorch deep learning model for fine-grained head pose estimation from images and video, without requiring facial keypoints. I… | 32 | 1666 | maintenance |
| NanoTrack A PyTorch collection of Siamese-based visual object tracking models including SiamFC, SiamRPN++, SiamMask, Ocean, LightTrack, and the light… | 32 | 1660 | maintenance |
| rwightman/efficientdet-pytorch A PyTorch implementation of EfficientDet object detection, faithful to the original Google TensorFlow implementation with ported pretrained… | 23 | 1656 | maintenance |
| vlfeat/vlfeat VLFeat is an open-source C library of popular computer vision algorithms specializing in image understanding and local feature extraction a… | 32 | 1650 | maintenance |
| invictus717/MetaTransformer Meta-Transformer is a research framework for unified multimodal learning that maps inputs from 12 modalities (text, images, point clouds, a… | 20 | 1643 | maintenance |
| facebookresearch/consistent_depth A research library from Facebook AI Research implementing Consistent Video Depth Estimation (SIGGRAPH 2020). It reconstructs dense, flicker… | 10 | 1634 | maintenance |
| raulmur/ORB_SLAM ORB-SLAM is a real-time monocular SLAM system written in C++ that computes camera trajectories and sparse 3D reconstructions from a single … | 32 | 1633 | maintenance |
| uzh-rpg/rpg_svo_pro_open SVO Pro is a C++ implementation of Semi-direct Visual Odometry from the Robotics and Perception Group at University of Zurich, supporting m… | 32 | 1613 | maintenance |
| DT42/BerryNet BerryNet is a deep learning gateway that turns edge devices like Raspberry Pi into intelligent, offline AI hubs for analyzing camera images… | 23 | 1611 | maintenance |
| experiencor/keras-yolo3 A Keras/TensorFlow implementation of YOLOv3 for object detection, supporting detection with pretrained weights, custom model training with … | 32 | 1608 | maintenance |
| ialhashim/DenseDepth Official Keras/TensorFlow implementation (with experimental PyTorch and TF2 code) of the DenseDepth paper for high-quality monocular depth … | 32 | 1607 | maintenance |
| wy1iu/sphereface Official implementation of SphereFace (Deep Hypersphere Embedding for Face Recognition, CVPR 2017), built on Caffe with a full face recogni… | 32 | 1606 | maintenance |
| jiupinjia/stylized-neural-painting Official PyTorch implementation of the CVPR 2021 paper 'Stylized Neural Painting', which translates photos into vectorized painting artwork… | 32 | 1605 | maintenance |
| chandrikadeb7/Face-Mask-Detection A face mask detection system built with OpenCV and TensorFlow/Keras that uses deep learning (SSD MobileNetV2) to detect whether people are … | 23 | 1605 | maintenance |
| Fast AutoAugment Official PyTorch implementation of Fast AutoAugment (NeurIPS 2019), which learns image augmentation policies via density-matching search. I… | 32 | 1604 | maintenance |
| PeterWang512/FALdetector FALdetector is the official PyTorch implementation of the ICCV 2019 paper 'Detecting Photoshopped Faces by Scripting Photoshop'. It provide… | 32 | 1601 | maintenance |
| cvg/nice-slam NICE-SLAM is a dense RGB-D SLAM system that combines neural implicit decoders with hierarchical grid-based scene representations, published… | 32 | 1600 | maintenance |
| silvanmelchior/RPi_Cam_Web_Interface A PHP-based web interface for controlling the Raspberry Pi Camera module. It provides live streaming, motion detection, time-lapse capture,… | 32 | 1599 | maintenance |
| niessner/BundleFusion BundleFusion is a real-time, globally consistent 3D reconstruction system for RGB-D input, published at SIGGRAPH 2017. It estimates globall… | 32 | 1590 | maintenance |
| NVlabs/FUNIT FUNIT is NVIDIA's PyTorch implementation of a few-shot unsupervised image-to-image translation model (ICCV 2019) that can translate images … | 32 | 1586 | maintenance |
| lufficc/SSD A high-quality, fast, modular reference implementation of the SSD (Single Shot MultiBox Detector) object detection model in PyTorch. It sup… | 23 | 1585 | maintenance |
| melodyguan/enas Authors' TensorFlow implementation of Efficient Neural Architecture Search (ENAS), which discovers neural network architectures via paramet… | 32 | 1579 | maintenance |
| snavely/bundler_sfm Bundler is a structure-from-motion (SfM) system that takes unordered image collections with features and matches and produces a 3D reconstr… | 32 | 1579 | maintenance |
| vturrisi/solo-learn solo-learn is a Python library of state-of-the-art self-supervised methods for unsupervised visual representation learning, built on PyTorc… | 62 | 1578 | maintenance |
| xinntao/EDVR EDVR is the winning solution of the NTIRE19 video restoration challenges, built on enhanced deformable convolutional networks. The repo is … | 32 | 1578 | maintenance |
| Temporal Segment Networks (TSN) Official code and pretrained models for Temporal Segment Networks (TSN), a deep learning framework for video action recognition published a… | 32 | 1577 | maintenance |
| Ewenwan/ORB_SLAM2_SSD_Semantic A C++ research project extending ORB_SLAM2 with dynamic object detection and semantic mapping. It combines SSD-based object detection (Mobi… | 32 | 1575 | maintenance |
| Azure Kinect SDK A cross-platform (Linux and Windows) user-mode SDK in C++ for reading data from the Azure Kinect depth and RGB camera device. It provides a… | 10 | 1575 | maintenance |
| kumar-shridhar/PyTorch-BayesianCNN A PyTorch library implementing Bayesian convolutional neural networks with variational inference via Bayes by Backprop. It provides drop-in… | 32 | 1574 | maintenance |
| sniklaus/3d-ken-burns A PyTorch reference implementation of the 3D Ken Burns Effect from a Single Image paper, which animates a still photo with a virtual camera… | 67 | 1572 | maintenance |