Open-source deep-learning software
2663 open-source libraries, tools and applications tagged deep-learning, ranked by how closely deep-learning 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.
2663 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| facebookresearch/vggsfm VGGSfM is a deep learning-based Structure from Motion pipeline from Meta AI and Oxford VGG that recovers camera poses and 3D point clouds f… | 26 | 1430 | active |
| CSAILVision/semantic-segmentation-pytorch A PyTorch implementation of semantic segmentation (scene parsing) models for the MIT ADE20K dataset, including pretrained model zoo and tra… | 32 | 5081 | maintenance |
| SpikingBrain SpikingBrain-7B is a brain-inspired large language model that combines hybrid efficient attention, MoE modules, and spike encoding, with a … | 51 | 1408 | active |
| om-ai-lab/OmDet OmDet-Turbo is a PyTorch implementation of a transformer-based open-vocabulary object detection model that detects arbitrary user-defined o… | 54 | 1406 | active |
| yfeng95/PRNet PRNet is a Python/TensorFlow implementation of the ECCV 2018 Position Map Regression Network for joint 3D face reconstruction and dense ali… | 32 | 5018 | maintenance |
| ByteDance-Seed/SeedVR SeedVR/SeedVR2 are diffusion-transformer based models for generic real-world and AIGC video and image restoration, with SeedVR2 using adver… | 44 | 1390 | active |
| QwenAudio/ThinkSound ThinkSound is a PyTorch implementation of a NeurIPS 2025 framework that generates and edits audio from video, text, or audio inputs using C… | 49 | 1380 | active |
| siyuanliii/masa Official PyTorch implementation of MASA (CVPR 2024 Highlight), a universal instance appearance model that learns to match any objects acros… | 30 | 1376 | active |
| yinguobing/head-pose-estimation A Python library for realtime human head pose estimation using ONNX Runtime and OpenCV. It combines face detection (SCRFD), 68-point facial… | 23 | 1357 | stable |
| lxtGH/OMG-Seg Official research codebase for OMG-Seg (CVPR 2024) and OMG-LLaVA (NeurIPS 2024), unified models for image-level, object-level, and pixel-le… | 68 | 1353 | active |
| owlbarn/owl Owl is an OCaml library for scientific and engineering computing, providing n-dimensional arrays, linear algebra, statistics, optimization,… | 62 | 1351 | active |
| bytedance/Bernini Bernini is a unified framework for video generation and editing that combines an MLLM-based semantic planner with a DiT-based renderer perf… | 54 | 1330 | active |
| Vahe1994/AQLM Official PyTorch implementation of AQLM, an extreme LLM compression method via additive quantization, extended with PV-Tuning for finetunin… | 54 | 1329 | active |
| Parskatt/RoMa RoMa (romatch) is a Python library for robust dense feature matching between image pairs, estimating pixel-dense warps and reliable certain… | 46 | 1309 | active |
| seetaface/SeetaFaceEngine SeetaFace Engine is an open-source C++ face recognition engine comprising face detection, face alignment, and face identification modules. … | 32 | 4638 | maintenance |
| zju3dv/MatchAnything MatchAnything is a deep learning model for universal cross-modality image matching, released as research code accompanying a TPAMI 2026 pap… | 64 | 1293 | active |
| city-super/Scaffold-GS Scaffold-GS is a research implementation of a structured 3D Gaussian splatting method that uses anchor points on a sparse voxel grid to dis… | 27 | 1286 | active |
| DreamLM/Dream Dream 7B is an open diffusion large language model (dLLM) with base and instruct checkpoints, plus inference and training code built on Hug… | 40 | 1279 | active |
| Tianxiaomo/pytorch-YOLOv4 A minimal PyTorch implementation of YOLOv4 (and YOLOv4-tiny) supporting inference and training, with tools to convert Darknet weights to Py… | 32 | 4520 | maintenance |
| lpiccinelli-eth/UniDepth UniDepth is a Python library and research codebase for universal monocular metric depth estimation from single images, based on CVPR 2024 a… | 32 | 1274 | active |
| ZHKKKe/MODNet MODNet is a deep learning model for real-time portrait matting (background removal) that requires only an RGB image as input, with no trima… | 32 | 4376 | maintenance |
| sh-lee-prml/HierSpeechpp Official PyTorch implementation of HierSpeech++, a fast zero-shot speech synthesizer for text-to-speech and voice conversion based on hiera… | 29 | 1237 | active |
| Soul-AILab/SoulX-LiveAct SoulX-LiveAct is the official inference code for a real-time human animation framework that generates lifelike, audio/multimodal-controlled… | 51 | 1193 | active |
| wladradchenko/wunjo.wladradchenko.ru Wunjo CE is an open-source, locally-run AI media suite for face swap, lip sync, voice cloning, object/text/background removal, restyling, a… | 84 | 1173 | active |
| NVlabs/imaginaire NVIDIA's PyTorch library containing optimized implementations of image and video synthesis methods, including GAN-based image-to-image tran… | 32 | 4079 | maintenance |
| FoundationVision/GLEE GLEE is a general object foundation model for images and videos that unifies detection, segmentation, tracking, grounding, and open-world o… | 27 | 1171 | active |
| cure-lab/MagicDrive MagicDrive is the official PyTorch implementation of an ICLR 2024 paper for controllable street view generation using diffusion models with… | 32 | 1165 | active |
| facebookresearch/fairseq2 fairseq2 is a PyTorch-based sequence modeling toolkit from Meta FAIR for training custom models for content generation tasks such as langua… | 86 | 1152 | active |
| InternRobotics/InternNav InternNav is an open-source PyTorch-based toolbox for building embodied navigation foundation models, supporting vision-language navigation… | 52 | 1135 | active |
| open-mmlab/mmtracking MMTracking is OpenMMLab's PyTorch-based toolbox for video perception tasks, unifying video object detection, multiple object tracking, sing… | 23 | 3903 | maintenance |
| lasgroup/SDPO SDPO (Self-Distilled Policy Optimization) is a research library implementing a reinforcement learning framework for post-training large lan… | 53 | 1126 | active |
| hkchengrex/Cutie Cutie is a video object segmentation framework with object-level memory reading, a follow-up to XMem offering better consistency, robustnes… | 18 | 1122 | active |
| princeton-vl/RAFT-Stereo RAFT-Stereo is a PyTorch implementation of a deep learning model for stereo matching that estimates disparity maps from stereo image pairs … | 71 | 1121 | stable |
| yandex/YaLM-100B YaLM-100B is a GPT-like pretrained language model with 100 billion parameters, trained by Yandex on English and Russian text using DeepSpee… | 32 | 3756 | maintenance |
| microsoft/Biodiversity Microsoft AI for Good Lab's biodiversity research hub providing open-source AI models and tools for wildlife monitoring and conservation, i… | 86 | 1086 | active |
| LujiaJin/One-Pot_Multi-Frame_Denoising Official PyTorch implementation of the One-Pot Multi-frame Denoising (OPD) method published at BMVC 2022 and extended in IJCV. It provides … | 57 | 1084 | stable |
| FINN FINN is an open-source dataflow compiler from AMD/Xilinx that generates highly efficient FPGA accelerators for quantized neural network (QN… | 85 | 1077 | active |
| zai-org/GLM-TTS GLM-TTS is a Python-based text-to-speech synthesis system built on large language models, using a two-stage LLM plus Flow model architectur… | 47 | 1066 | active |
| qqlu/Entity EntitySeg is an open-source PyTorch toolbox for open-world, high-quality image segmentation, built on Detectron2. It aggregates multiple re… | 32 | 1049 | active |
| williamyang1991/VToonify Official PyTorch implementation of VToonify, a SIGGRAPH Asia 2022 framework for controllable high-resolution portrait video style transfer … | 32 | 3579 | maintenance |
| Tencent-Hunyuan/InstantCharacter InstantCharacter is a tuning-free framework built on diffusion transformers that generates character-consistent images from a single refere… | 26 | 1044 | active |
| ml-tooling/ml-workspace ML Workspace is an all-in-one web-based IDE Docker image specialized for machine learning and data science. It bundles Jupyter, JupyterLab,… | 23 | 3546 | maintenance |
| kuleshov-group/bd3lms BD3-LMs is a research implementation of Block Discrete Denoising Diffusion Language Models that interpolate between autoregressive and diff… | 31 | 1037 | active |
| Jumpat/SegmentAnythingin3D SA3D is a research framework that lifts 2D Segment Anything (SAM) masks into 3D segmentation of objects within a NeRF or 3D Gaussian Splatt… | 37 | 1031 | active |
| zhyever/PatchFusion PatchFusion is a CVPR 2024 end-to-end tile-based framework for high-resolution monocular metric depth estimation from single images. It fus… | 54 | 1025 | active |
| microsoft/tensorwatch TensorWatch is a Python library from Microsoft Research for debugging, monitoring, and visualizing machine learning training in real time, … | 62 | 3478 | maintenance |
| yangxy/PASD PASD (Pixel-Aware Stable Diffusion) is a Python research codebase implementing an ECCV 2024 method for realistic image super-resolution and… | 28 | 1022 | active |
| HumeAI/tada TADA is an open-source speech-language model from Hume AI that generates expressive, high-fidelity speech via text-acoustic dual alignment,… | 48 | 1020 | active |
| eragonruan/text-detection-ctpn A TensorFlow implementation of the Connectionist Text Proposal Network (CTPN) for detecting horizontal scene text in images. It includes pr… | 23 | 3427 | maintenance |
| facebookresearch/Mask2Former Mask2Former is the official PyTorch implementation of the CVPR 2022 paper 'Masked-attention Mask Transformer for Universal Image Segmentati… | 10 | 3417 | maintenance |
| clovaai/CRAFT-pytorch Official PyTorch implementation of CRAFT (Character Region Awareness for Text Detection), a scene text detector that localizes text by pred… | 32 | 3406 | maintenance |
| minimaxir/gpt-2-simple A Python package that simplifies fine-tuning OpenAI's GPT-2 text-generation model (124M/355M) on custom text and generating text from the r… | 23 | 3405 | maintenance |
| bytedance/lightseq LightSeq is a high-performance CUDA-based library for training and inference of sequence models like Transformer, BERT, GPT, and BART, with… | 10 | 3292 | maintenance |
| google-research/albert Official TensorFlow implementation and pretrained checkpoints of ALBERT, a lite version of BERT for self-supervised learning of language re… | 10 | 3278 | maintenance |
| anandpawara/Real_Time_Image_Animation A real-time Python application that animates a still image (e.g., a portrait) using facial motion from a live camera or video file, built o… | 32 | 3246 | maintenance |
| mkocabas/VIBE Official PyTorch implementation of VIBE (CVPR 2020), a video-based method for 3D human body pose and shape estimation that predicts SMPL bo… | 23 | 3210 | maintenance |
| google-research/frame-interpolation FILM is the official TensorFlow 2 implementation of a state-of-the-art frame interpolation neural network from Google Research, presented a… | 10 | 3153 | maintenance |
| dbiir/UER-py UER-py is a PyTorch framework for pre-training transformer language models (BERT, GPT-2, T5, ELMo, etc.) and fine-tuning them on downstream… | 32 | 3114 | maintenance |
| biubug6/Pytorch_Retinaface A PyTorch implementation of the RetinaFace single-stage face detection model, supporting mobilenet0.25 and resnet50 backbones with pretrain… | 32 | 2976 | maintenance |
| Tencent/FaceDetection-DSFD DSFD (Dual Shot Face Detector) is Tencent Youtu's high-accuracy face detection network, released with PyTorch inference code and pretrained… | 52 | 2975 | maintenance |
| Alpha-VLLM/LLaMA2-Accessory LLaMA2-Accessory is an open-source Python toolkit for pretraining, finetuning, and deploying large language models and multimodal LLMs, inc… | 29 | 2796 | maintenance |
| tensorflow/graphics TensorFlow Graphics is a library of differentiable graphics layers for TensorFlow, including differentiable renderers, spatial transformers… | 66 | 2782 | maintenance |
| YCG09/chinese_ocr An end-to-end Chinese OCR system implemented with TensorFlow and Keras, combining CTPN for text detection with DenseNet + CTC for text reco… | 32 | 2776 | maintenance |
| HypoX64/DeepMosaics DeepMosaics is a Python application that automatically removes or adds mosaics in images and videos using semantic segmentation and image-t… | 23 | 2659 | maintenance |
| PeterH0323/Smart_Construction A YOLOv5-based object detection application for detecting people, heads, and safety helmets on construction sites, including pretrained wei… | 23 | 2628 | maintenance |
| microsoft/GLIP GLIP is Microsoft's official implementation of Grounded Language-Image Pre-training, a vision-language model that unifies object detection … | 32 | 2612 | maintenance |
| zllrunning/face-parsing.PyTorch A PyTorch implementation of face parsing using a modified BiSeNet architecture, trained on the CelebAMask-HQ dataset. It provides training … | 32 | 2590 | maintenance |
| s3prl/s3prl S3PRL is a PyTorch toolkit for self-supervised speech pre-training and representation learning, bundling many upstream models like wav2vec … | 51 | 2567 | maintenance |
| yfeng95/DECA DECA is the official PyTorch implementation of a SIGGRAPH 2021 method that reconstructs a detailed 3D head model (pose, shape, facial detai… | 32 | 2524 | maintenance |
| leggedrobotics/darknet_ros A ROS package wrapping the YOLO (Darknet) real-time object detector for use in robotic systems. It subscribes to camera image topics and pu… | 23 | 2443 | maintenance |
| Hzzone/pytorch-openpose A PyTorch reimplementation of OpenPose for body and hand pose estimation, with models converted directly from the original OpenPose caffemo… | 32 | 2318 | maintenance |
| MhLiao/DB A PyTorch implementation of DBNet and DBNet++, real-time arbitrary-shape scene text detection models based on differentiable binarization. … | 32 | 2267 | maintenance |
| hustvl/YOLOP YOLOP is a multi-task deep learning network that jointly performs traffic object detection, drivable area segmentation, and lane detection … | 32 | 2249 | maintenance |
| XLabs AI Flux Toolkit XLabs AI's training scripts for fine-tuning the FLUX.1 diffusion model with LoRA, ControlNet, and IP-Adapter adapters, using DeepSpeed and … | 24 | 2227 | maintenance |
| mit-han-lab/temporal-shift-module PyTorch implementation of the Temporal Shift Module (TSM), an ICCV 2019 technique that adds temporal modeling to 2D CNNs at zero extra comp… | 32 | 2223 | maintenance |
| alibaba/EasyNLP EasyNLP is a comprehensive PyTorch-based NLP toolkit from Alibaba that provides training, inference, and deployment for pre-trained languag… | 23 | 2186 | maintenance |
| bubbliiiing/yolov4-pytorch A PyTorch implementation of the YOLOv4 object detection model with full training, prediction, and evaluation scripts. It supports training … | 23 | 2163 | maintenance |
| facebookresearch/ELF ELF is an end-to-end, lightweight and flexible C++/Python platform for game research, focused on real-time strategy games. It hosts multipl… | 10 | 2092 | maintenance |
| jiupinjia/SkyAR SkyAR is the official PyTorch implementation of the paper 'Castle in the Sky: Dynamic Sky Replacement and Harmonization in Videos'. It perf… | 32 | 2029 | maintenance |
| facebookresearch/Detic Detic is the official code release for the ECCV 2022 paper 'Detecting Twenty-thousand Classes using Image-level Supervision'. It is an open… | 32 | 2013 | maintenance |
| Music-and-Culture-Technology-Lab/omnizart Omnizart is a Python library and CLI for automatic music transcription, transcribing pitched instruments, vocal melody, chords, drum events… | 82 | 1979 | maintenance |
| yehengchen/Object-Detection-and-Tracking A collection of Python implementations combining YOLO-based object detection with SORT and DeepSORT multi-object tracking. It includes exam… | 32 | 1965 | maintenance |
| mpatacchiola/deepgaze Deepgaze is a Python computer vision library for human-computer interaction built on OpenCV and TensorFlow. It provides CNN-based head pose… | 32 | 1882 | 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 |
| 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 |
| 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 |
| 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 |
| deepset-ai/FARM FARM is a Python framework for fine-tuning and evaluating transformer-based language models for NLP tasks, with a focus on question answeri… | 10 | 1753 | 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 |
| 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 |
| 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 |
| lucidrains/PaLM-rlhf-pytorch A PyTorch library implementing Reinforcement Learning from Human Feedback (RLHF) on top of the PaLM transformer architecture, aiming to rep… | 77 | 7861 | experimental |
| 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 |
| PeterWang512/FALdetector FALdetector is the official PyTorch implementation of the ICCV 2019 paper 'Detecting Photoshopped Faces by Scripting Photoshop'. It provide… | 32 | 1601 | maintenance |
| 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 |
| 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 |
| 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 |
| FudanNLP/fitlog fitlog (fast + git + log) is a Python tool that helps deep learning practitioners record training logs and manage experiment code, combinin… | 23 | 1511 | 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 |
| 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 |