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 |
|---|---|---|---|
| clovaai/voxceleb_trainer A PyTorch framework for training and evaluating speaker recognition and verification models on the VoxCeleb datasets. It implements multipl… | 64 | 1176 | maintenance |
| ermongroup/SDEdit SDEdit is a PyTorch implementation of the ICLR 2022 paper 'Guided Image Synthesis and Editing with Stochastic Differential Equations'. It p… | 32 | 1173 | maintenance |
| xiangwang1223/knowledge_graph_attention_network TensorFlow implementation of KGAT (Knowledge Graph Attention Network), a KDD 2019 paper on knowledge-aware personalized recommendation. It … | 32 | 1167 | maintenance |
| alexandre01/deepsvg Official PyTorch code for the NeurIPS 2020 paper 'DeepSVG: A Hierarchical Generative Network for Vector Graphics Animation'. It provides a … | 74 | 1166 | maintenance |
| shadow2496/VITON-HD Official PyTorch implementation of VITON-HD (CVPR 2021), a high-resolution image-based virtual try-on model that transfers a clothing item … | 37 | 1166 | maintenance |
| facebookresearch/meshrcnn Mesh R-CNN is Facebook AI Research's official implementation of the ICCV 2019 paper, a model that detects objects in images and predicts th… | 56 | 1162 | maintenance |
| spring-media/TransformerTTS A TensorFlow 2 implementation of a non-autoregressive Transformer-based neural network for text-to-speech synthesis, based on FastSpeech an… | 10 | 1161 | maintenance |
| cheungdaven/DeepRec DeepRec is an open-source Python/TensorFlow toolkit implementing deep learning based recommendation models for rating prediction, top-N ite… | 32 | 1160 | maintenance |
| wyharveychen/CloserLookFewShot Reference PyTorch implementation of the ICLR 2019 paper 'A Closer Look at Few-shot Classification', providing a testbed for empirical evalu… | 32 | 1160 | maintenance |
| Kyubyong/dc_tts A TensorFlow implementation of DC-TTS, a text-to-speech model based on deep convolutional networks with guided attention. It includes train… | 32 | 1160 | maintenance |
| ZhuiyiTechnology/roformer RoFormer is an MLM pre-trained language model built on rotary position embeddings (RoPE), a relative position encoding method with strong t… | 32 | 1155 | maintenance |
| Xharlie/pointnerf Point-NeRF is a research implementation of a point-based neural radiance field method (CVPR 2022 Oral) that models scenes with neural 3D po… | 32 | 1154 | maintenance |
| houqb/CoordAttention A PyTorch implementation of Coordinate Attention (CoordAttention), an efficient attention mechanism for mobile networks introduced in a CVP… | 32 | 1154 | maintenance |
| abhishekkrthakur/tez Tez is a lightweight, simple trainer library for PyTorch that simplifies training loops while keeping users close to native PyTorch. It sup… | 23 | 1153 | maintenance |
| bubbliiiing/yolov5-pytorch A PyTorch implementation of the YOLOv5 (v5.0) object detection model with full training, prediction, and evaluation pipelines. It is design… | 23 | 1152 | maintenance |
| HackerPoet/Composer Composer is a neural network-based application that generates video game music. It is written in Python and produces chiptune-style game so… | 32 | 1148 | maintenance |
| mchong6/GANsNRoses Official PyTorch implementation of GANs N' Roses, a model for diverse image-to-image and video-to-video translation that converts selfie ph… | 32 | 1148 | maintenance |
| tech-srl/code2vec Official TensorFlow implementation of the code2vec model from the POPL'2019 paper, which learns distributed vector representations of code … | 32 | 1148 | maintenance |
| braindotai/Watermark-Removal-Pytorch A PyTorch implementation of watermark removal based on the Deep Image Prior paper, using a CNN generator's structure to restore images with… | 23 | 1148 | maintenance |
| tensorflow/benchmarks A benchmark framework for TensorFlow, containing PerfZero for running and reporting TensorFlow benchmarks and legacy CNN benchmark scripts … | 10 | 1148 | maintenance |
| showlab/Show-1 Show-1 is a research codebase implementing a text-to-video generation model that combines pixel and latent diffusion models, published at I… | 43 | 1147 | maintenance |
| joe-siyuan-qiao/DetectoRS Official PyTorch implementation of DetectoRS, a state-of-the-art object detection and instance segmentation model using Recursive Feature P… | 32 | 1146 | maintenance |
| pix2pixzero/pix2pix-zero pix2pix-zero is a Python library implementing zero-shot image-to-image translation using pre-trained Stable Diffusion models. It enables ed… | 31 | 1146 | maintenance |
| HRNet/HRNet-Facial-Landmark-Detection Official PyTorch implementation of HRNet-based facial landmark detection from the TPAMI paper 'Deep High-Resolution Representation Learning… | 32 | 1140 | maintenance |
| patil-suraj/question_generation An open-source study and library for neural question generation using pre-trained seq2seq transformer models like T5 via Hugging Face trans… | 32 | 1138 | maintenance |
| arpitbansal297/Cold-Diffusion-Models Official PyTorch implementation of Cold Diffusion, a research paper showing that diffusion-style generative models can invert arbitrary ima… | 32 | 1137 | maintenance |
| Timthony/self_drive A self-driving RC car project based on Raspberry Pi and TensorFlow/Keras. It collects camera images while a human drives the car on a taped… | 32 | 1137 | maintenance |
| google-research/mixmatch Reference implementation of MixMatch, a holistic semi-supervised learning algorithm from a Google Research paper. It includes dataset prepa… | 10 | 1136 | maintenance |
| karfly/learnable-triangulation-pytorch Official PyTorch implementation of the ICCV 2019 paper 'Learnable Triangulation of Human Pose', providing algebraic and volumetric learnabl… | 23 | 1135 | maintenance |
| harvardnlp/pytorch-struct Torch-Struct is a PyTorch library of tested, GPU-accelerated implementations of core structured prediction algorithms such as CRFs, HMMs, H… | 23 | 1134 | maintenance |
| youyuge34/Anime-InPainting An application tool built on Edge-Connect that performs anime image inpainting, allowing users to repair, fill, and remove blemishes or mos… | 32 | 1133 | maintenance |
| fff-rs/juice Juice is a Rust machine learning framework ('The Hacker's Machine Learning Engine') built on the Coaster hardware abstraction layer, suppor… | 32 | 1132 | maintenance |
| Cloud-CV/Fabrik Fabrik is a web-based collaborative IDE for building, visualizing, and designing neural networks via a drag-and-drop interface. It supports… | 32 | 1131 | maintenance |
| maudzung/SFA3D A PyTorch implementation of SFA3D, a fast and accurate anchor-free 3D object detection model for LiDAR point clouds, trained and evaluated … | 32 | 1131 | maintenance |
| yu-takagi/StableDiffusionReconstruction Research codebase reproducing Takagi and Nishimoto's CVPR 2023 method for reconstructing images a person viewed from fMRI brain activity us… | 31 | 1130 | maintenance |
| KinWaiCheuk/nnAudio nnAudio is a PyTorch-based audio processing toolbox that computes spectrograms (STFT, Mel, CQT, MFCC) using 1D convolutional neural network… | 57 | 1129 | maintenance |
| MRzzm/DINet DINet is the official PyTorch implementation of an AAAI 2023 paper on realistic face visually dubbing, which deforms and inpaints mouth reg… | 32 | 1129 | maintenance |
| bearpaw/pytorch-pose A PyTorch toolkit implementing a general pipeline for 2D single-human pose estimation, with training, inference, and evaluation interfaces … | 32 | 1120 | maintenance |
| irolaina/FCRN-DepthPrediction Reference implementation and pretrained models for FCRN (Deeper Depth Prediction with Fully Convolutional Residual Networks), predicting de… | 32 | 1119 | maintenance |
| thuml/TimesNet TimesNet is the official code release for the ICLR 2023 paper on temporal 2D-variation modeling for general time series analysis. It transf… | 31 | 1118 | maintenance |
| Harmonai-org/sample-generator A set of tools and Jupyter notebooks for training generative diffusion models on arbitrary audio samples, built around Dance Diffusion. It … | 32 | 1116 | maintenance |
| hotshotco/Hotshot-XL Hotshot-XL is an AI text-to-GIF model built to work alongside Stable Diffusion XL, generating 1-second GIFs at 8 FPS. It supports any fine-… | 28 | 1116 | maintenance |
| snap-research/EfficientFormer A PyTorch implementation of EfficientFormer and EfficientFormerV2, efficient vision transformer model families designed to run at MobileNet… | 32 | 1115 | maintenance |
| mit-han-lab/gan-compression A PyTorch research codebase implementing GAN Compression, a CVPR 2020 method that reduces the computation of conditional GANs like pix2pix,… | 23 | 1115 | maintenance |
| Res2Net/Res2Net-PretrainedModels Official PyTorch implementation of Res2Net, a multi-scale CNN backbone architecture published in TPAMI, with ImageNet-pretrained model weig… | 32 | 1114 | maintenance |
| andyzeng/visual-pushing-grasping PyTorch reference implementation of Visual Pushing and Grasping (VPG), which trains robotic agents via self-supervised deep reinforcement l… | 32 | 1113 | maintenance |
| rosinality/style-based-gan-pytorch A PyTorch implementation of the StyleGAN paper (A Style-Based Generator Architecture for Generative Adversarial Networks), including traini… | 32 | 1110 | maintenance |
| baidu-research/DeepBench DeepBench is a benchmarking suite from Baidu Research that measures the performance of fundamental deep learning operations (matrix multipl… | 32 | 1109 | maintenance |
| fudan-zvg/SETR SETR (SEgmentation TRansformers) is the official PyTorch implementation of the CVPR 2021 / IJCV 2024 paper 'Rethinking Semantic Segmentatio… | 32 | 1109 | maintenance |
| novak-99/MLPP ML++ is a C++ machine learning library providing regression models, dynamically sized neural networks, statistics, and optimization algorit… | 23 | 1109 | maintenance |
| qiucheng025/zao- A Python deep learning tool that identifies and swaps faces in images and videos, with extract, train, and convert workflows plus an option… | 32 | 1108 | maintenance |
| aim-uofa/AdelaiDepth AdelaiDepth is an open-source toolbox for monocular depth prediction and 3D scene reconstruction from single images, containing research pr… | 32 | 1106 | maintenance |
| msracver/Relation-Networks-for-Object-Detection Official MXNet implementation of the CVPR 2018 paper 'Relation Networks for Object Detection', which adds an attention-based relation modul… | 32 | 1104 | maintenance |
| keroro824/HashingDeepLearning SLIDE is a C++ research codebase implementing locality-sensitive hashing based training of deep neural networks, from the paper 'In Defense… | 32 | 1104 | maintenance |
| PaddlePaddle/Paddle.js Paddle.js is a browser-based deep learning inference engine for Baidu PaddlePaddle models, running via WebGL, WebGPU, or WebAssembly backen… | 23 | 1104 | maintenance |
| pfnet-research/sngan_projection Official Chainer implementation of GANs using spectral normalization and projection discriminators for conditional image generation on Imag… | 32 | 1102 | maintenance |
| fyu/drn A PyTorch library implementing Dilated Residual Networks (DRN), which combine dilated convolutions with residual networks for image classif… | 32 | 1102 | maintenance |
| hongfz16/AvatarCLIP Official PyTorch implementation of AvatarCLIP, a SIGGRAPH 2022 research framework that generates and animates 3D human avatars from natural… | 32 | 1101 | maintenance |
| kazuto1011/deeplab-pytorch An unofficial PyTorch re-implementation of DeepLab v2 with a ResNet-101 backbone for semantic segmentation, supporting COCO-Stuff and PASCA… | 23 | 1100 | maintenance |
| hhaAndroid/mmdetection-mini A minimal, heavily annotated reimplementation of the mmdetection object detection framework, built from scratch for learning purposes. It m… | 32 | 1099 | maintenance |
| jacobgil/vit-explain A PyTorch library implementing explainability methods for Vision Transformers, including Attention Rollout and Gradient Attention Rollout. … | 32 | 1099 | maintenance |
| VITA-Group/EnlightenGAN Official PyTorch implementation of EnlightenGAN, an IEEE TIP paper on deep low-light image enhancement without paired supervision using gen… | 32 | 1095 | maintenance |
| NVlabs/NVAE Official PyTorch implementation of NVAE, a deep hierarchical variational autoencoder for likelihood-based image generation from a NeurIPS 2… | 32 | 1094 | maintenance |
| Tencent/TencentPretrain TencentPretrain is a PyTorch-based toolkit for pre-training and fine-tuning transformer models across modalities such as text, vision, and … | 32 | 1094 | maintenance |
| maelfabien/Multimodal-Emotion-Recognition A real-time multimodal emotion recognition web app built with Flask that analyzes emotions from text, audio, and video inputs using deep le… | 32 | 1093 | maintenance |
| yu4u/noise2noise An unofficial, partial Keras implementation of the Noise2Noise paper for learning image restoration without clean training data. It trains … | 23 | 1089 | maintenance |
| ethanhe42/channel-pruning Reference implementation of the ICCV 2017 channel pruning method for accelerating very deep convolutional neural networks, using LASSO regr… | 23 | 1088 | maintenance |
| Zero-DCE PyTorch implementation of Zero-DCE (Zero-Reference Deep Curve Estimation), a CVPR 2020 model that enhances low-light images without paired … | 32 | 1087 | maintenance |
| sfzhang15/ATSS Official PyTorch implementation of ATSS (Adaptive Training Sample Selection), a CVPR 2020 Oral paper on object detection. It automatically … | 32 | 1085 | maintenance |
| locuslab/convmixer An official PyTorch implementation of ConvMixer, a patch-based convolutional image classification architecture from the ICLR 2022 paper 'Pa… | 23 | 1085 | maintenance |
| GOATmessi8/ASFF A PyTorch implementation of YOLOv3 with the Adaptively Spatial Feature Fusion (ASFF) module and optional MobileNetV2 backbone for single-sh… | 32 | 1084 | maintenance |
| wilson1yan/VideoGPT VideoGPT is a PyTorch implementation of a video generation model that combines a VQ-VAE with 3D convolutions and axial attention with a GPT… | 32 | 1084 | maintenance |
| google-research/maxim Official implementation of MAXIM, a multi-axis MLP architecture for low-level image processing tasks, presented as a CVPR 2022 oral paper. … | 10 | 1084 | maintenance |
| weiyithu/SurroundOcc SurroundOcc is the official PyTorch implementation of an ICCV 2023 paper predicting dense 3D volumetric occupancy from multi-camera images … | 40 | 1083 | maintenance |
| rykov8/ssd_keras A Keras port of the SSD (Single Shot MultiBox Detector) object detection model, including pretrained weights ported from the original Caffe… | 10 | 1083 | maintenance |
| seongjunyun/Graph_Transformer_Networks Authors' PyTorch implementation of Graph Transformer Networks (GTN) from the NeurIPS 2019 paper, plus FastGTN with non-local operations. It… | 32 | 1081 | maintenance |
| AstarLight/CPS-OCR-Engine A deep-learning-based OCR engine from SYSU DeepDriving Lab that recognizes 3755 printed Chinese characters (Level-1 character set) in elect… | 32 | 1081 | maintenance |
| rixwew/pytorch-fm A PyTorch library implementing a wide collection of factorization machine models (FM, FFM, DeepFM, xDeepFM, AutoInt, NFM, DCN, and more) fo… | 32 | 1081 | maintenance |
| orobix/Prototypical-Networks-for-Few-shot-Learning-PyTorch A PyTorch implementation of Prototypical Networks for few-shot learning, including the prototypical loss and episode-based batch sampler. I… | 32 | 1078 | maintenance |
| pytorch/torchdynamo TorchDynamo is a Python-level JIT compiler that speeds up unmodified PyTorch programs by capturing Python bytecode into FX graphs. The proj… | 10 | 1078 | maintenance |
| floodsung/LearningToCompare_FSL Official PyTorch implementation of the Relation Network for Few-Shot Learning from the CVPR 2018 paper 'Learning to Compare'. It includes t… | 32 | 1076 | maintenance |
| DirectVoxGO DirectVoxGO (DVGO) is a PyTorch implementation of Direct Voxel Grid Optimization for fast neural radiance field (NeRF) reconstruction, repl… | 32 | 1075 | maintenance |
| JosephKJ/OWOD OWOD is the official PyTorch implementation of ORE (Open World Object Detector), a CVPR 2021 Oral paper introducing the Open World Object D… | 32 | 1073 | maintenance |
| YudongGuo/AD-NeRF A PyTorch implementation of AD-NeRF, an ICCV 2021 paper that synthesizes talking-head videos by driving neural radiance fields with audio i… | 32 | 1073 | maintenance |
| cardwing/Codes-for-Lane-Detection Reference implementations of lightweight lane detection CNNs, including the ENet-SAD model from the ICCV 2019 paper 'Learning Lightweight L… | 32 | 1072 | maintenance |
| megvii-research/PETR Official PyTorch implementation of PETR and PETRv2, transformer-based frameworks for multi-view 3D object detection and BEV segmentation fr… | 32 | 1071 | maintenance |
| gusye1234/LightGCN-PyTorch A PyTorch implementation of LightGCN, a simplified graph convolution network for collaborative filtering recommendation from the SIGIR 2020… | 10 | 1071 | maintenance |
| microsoft/VideoX VideoX is a collection of Microsoft's video cross-modal understanding models, including X-CLIP for video-language recognition, 2D-TAN and M… | 10 | 1071 | maintenance |
| MatthieuCourbariaux/BinaryNet BinaryNet is a research codebase for training deep neural networks whose weights and activations are constrained to +1 or -1, reproducing t… | 32 | 1068 | maintenance |
| omerbt/MultiDiffusion Official PyTorch implementation of MultiDiffusion (ICML 2023), a training-free framework that fuses multiple diffusion paths over a pre-tra… | 32 | 1068 | maintenance |
| NVIDIA-AI-IOT/trt_pose trt_pose is a Python library from NVIDIA for real-time human pose estimation accelerated with TensorRT, targeting NVIDIA Jetson and other N… | 23 | 1067 | maintenance |
| openai/blocksparse A Python package of TensorFlow ops and custom CUDA GPU kernels for block-sparse matrix multiplication and convolution, plus related ops lik… | 10 | 1067 | maintenance |
| deepmedic/deepmedic DeepMedic is an efficient multi-scale 3D convolutional neural network for segmenting 3D medical scans such as MRI and CT. It is a Python-ba… | 23 | 1066 | maintenance |
| jsyoon0823/TimeGAN Reference implementation of TimeGAN, a generative adversarial network framework for generating synthetic time-series data, published at Neu… | 58 | 1064 | maintenance |
| ternaus/TernausNet TernausNet is a PyTorch implementation of the U-Net architecture with a VGG11 encoder pre-trained on ImageNet for image segmentation. It wa… | 32 | 1062 | maintenance |
| OpenBMB/VisCPM VisCPM is a family of open-source bilingual (Chinese/English) multimodal large models built on the 10B CPM-Bee language model, comprising V… | 29 | 1062 | maintenance |
| RexYing/gnn-model-explainer Official implementation of GNNExplainer, a NeurIPS 2019 method for explaining predictions of graph neural networks by learning soft masks o… | 32 | 1060 | maintenance |
| OFA-Sys/ONE-PEACE ONE-PEACE is a general multimodal representation model that jointly encodes vision, audio, and language modalities without initializing fro… | 30 | 1060 | maintenance |
| philipperemy/keract Keract is a Python library that extracts layer activations (outputs) and gradients from TensorFlow/Keras models as NumPy arrays, with helpe… | 26 | 1060 | maintenance |
| wubinzzu/NeuRec NeuRec is an open-source Python library for recommender systems built on TensorFlow, offering 33 state-of-the-art neural recommendation mod… | 23 | 1060 | maintenance |