Open-source deep-learning software
2785 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.
2785 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| VITA-Group/DeblurGANv2 Official PyTorch implementation of DeblurGAN-v2, an ICCV 2019 relativistic conditional GAN for single-image motion deblurring with a Featur… | 32 | 1195 | maintenance |
| facebookresearch/mixup-cifar10 Official PyTorch implementation of mixup, a data augmentation technique that trains neural networks on convex combinations of image pairs a… | 10 | 1195 | maintenance |
| philipperemy/yolo-9000 A packaging of YOLO9000 (YOLOv2), a real-time object detection model that can detect over 9,000 object classes, built around the Darknet fr… | 32 | 1194 | maintenance |
| yitu-opensource/T2T-ViT Official PyTorch implementation of Tokens-to-Token ViT (T2T-ViT), an ICCV 2021 vision transformer architecture trained from scratch on Imag… | 23 | 1193 | maintenance |
| aitorzip/DeepGTAV DeepGTAV is a C++ plugin for Grand Theft Auto V that converts the game into a vision-based self-driving car research environment. It expose… | 32 | 1192 | maintenance |
| altosaar/variational-autoencoder A reference implementation of a variational autoencoder (VAE) in PyTorch, TensorFlow, and JAX, including an inverse autoregressive flow var… | 23 | 1192 | maintenance |
| jankrepl/deepdow deepdow is a Python library built on PyTorch that connects deep learning with portfolio optimization, allowing networks that output asset w… | 32 | 1190 | maintenance |
| elliottwu/unsup3d Official PyTorch implementation of the CVPR 2020 (Oral, Best Paper Award) research paper 'Unsupervised Learning of Probably Symmetric Defor… | 32 | 1189 | maintenance |
| DeepGCNs A PyTorch library implementing DeepGCNs, DeeperGCN, and GNN1000 - graph convolutional network architectures that borrow residual/dense conn… | 32 | 1189 | maintenance |
| vanhuyz/CycleGAN-TensorFlow A TensorFlow implementation of CycleGAN for unpaired image-to-image translation, based on the original CycleGAN paper. It includes scripts … | 23 | 1189 | maintenance |
| zju3dv/snake Official research code for 'Deep Snake for Real-Time Instance Segmentation' (CVPR 2020 oral), implementing a deep contour-based instance se… | 75 | 1188 | maintenance |
| lucidrains/performer-pytorch A PyTorch implementation of the Performer transformer, which uses linear attention via the FAVOR+ (Fast Attention Via positive Orthogonal R… | 23 | 1184 | maintenance |
| HyperGAN/HyperGAN HyperGAN is a composable GAN (generative adversarial network) framework built on PyTorch, offering both a Python API and a CLI with a user … | 68 | 1183 | maintenance |
| PyRetri/PyRetri PyRetri is a unified deep learning based unsupervised image retrieval toolbox built on PyTorch. It provides a modular pipeline of pre-proce… | 32 | 1181 | maintenance |
| Smerity/sha-rnn A PyTorch implementation of the Single Headed Attention RNN (SHA-LSTM) from Stephen Merity's paper, combining an LSTM with a single attenti… | 32 | 1180 | maintenance |
| naoto0804/pytorch-AdaIN An unofficial PyTorch implementation of the AdaIN paper 'Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization' (Huang… | 10 | 1180 | maintenance |
| julrog/nn_vis A Python tool that processes trained neural networks and renders them as decluttered 3D visualizations of their architecture and parameters… | 23 | 1179 | maintenance |
| fbcotter/pytorch_wavelets A PyTorch library implementing 2D discrete wavelet transforms (DWT), dual-tree complex wavelet transforms (DTCWT), their inverses, and a DT… | 23 | 1178 | maintenance |
| VoltaML/voltaML VoltaML is a lightweight Python library that compiles and optimizes machine learning and deep learning models for high-performance inferenc… | 32 | 1177 | maintenance |
| 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 |
| daniilidis-group/neural_renderer A PyTorch port of the Neural 3D Mesh Renderer (CVPR 2018), providing differentiable rasterization of 3D meshes. It supports rendering from … | 32 | 1170 | 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 |
| Neural 3D Mesh Renderer A differentiable 3D mesh renderer implemented in Chainer, accompanying the CVPR 2018 paper 'Neural 3D Mesh Renderer'. It enables gradient-b… | 23 | 1164 | 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 |
| ajay-sainy/Wav2Lip-GFPGAN A pipeline combining Wav2Lip lip-sync generation with GFPGAN face restoration to produce high-quality talking-head videos from an input vid… | 32 | 1159 | maintenance |
| Paperspace/DataAugmentationForObjectDetection A Python library implementing data augmentation transforms (flipping, scaling, rotation, shearing, translation, resizing) that adjust both … | 32 | 1156 | maintenance |
| Talking Head Anime from a Single Image Demo programs for the Talking Head Anime from a Single Image 2 project, which animates an anime character from a single image. It includes … | 32 | 1155 | 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 |
| 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 |
| uber-research/PPLM PPLM (Plug and Play Language Model) is a research implementation for controlled text generation that steers the topic and attributes of GPT… | 32 | 1152 | 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 |
| NVIDIA-Merlin/NVTabular NVTabular is a GPU-accelerated feature engineering and preprocessing library for tabular data, built to handle terabyte-scale datasets for … | 57 | 1151 | maintenance |
| PengyiZhang/SlimYOLOv3 SlimYOLOv3 is a research implementation of channel-pruned YOLOv3 object detectors optimized for real-time inference on UAVs and embedded de… | 32 | 1151 | 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 |
| 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 |
| peiyunh/tiny A MATLAB implementation of the Tiny Face Detector (CVPR 2017), a deep-learning face detector specialized in finding very small faces using … | 32 | 1143 | 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 |
| 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 |
| 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 |
| julvo/reloading A small Python library that reloads the source code of loops or functions on each iteration or invocation without losing runtime state. It … | 23 | 1129 | maintenance |
| microsoft/Windows-Machine-Learning Microsoft's Windows Machine Learning samples and tools repository, providing a high-performance ONNX inference API powered by ONNX Runtime … | 35 | 1124 | 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 |
| microsoft/MASS MASS is Microsoft's PyTorch implementation of Masked Sequence to Sequence Pre-training for language generation tasks. It provides pre-train… | 10 | 1117 | 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 |
| ckkelvinchan/RealBasicVSR Official PyTorch implementation of RealBasicVSR, a real-world video super-resolution model from CVPR 2022, built on MMEditing. It provides … | 32 | 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 |
| 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 |
| fudan-zvg/SETR SETR (SEgmentation TRansformers) is the official PyTorch implementation of the CVPR 2021 / IJCV 2024 paper 'Rethinking Semantic Segmentatio… | 32 | 1109 | 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 |
| 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 |
| vietnh1009/QuickDraw A Python implementation of Google's Quick, Draw! game that recognizes hand-drawn sketches using a CNN trained on the Quick Draw dataset. It… | 32 | 1104 | 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 |
| 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 |
| RUCAIBox/TextBox TextBox 2.0 is a Python/PyTorch library providing a unified pipeline for applying pre-trained language models to text generation tasks. It … | 23 | 1096 | 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 |
| 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 |
| Rock-100/FaceKit FaceKit is a C++ toolkit implementing Progressive Calibration Networks (PCN) for real-time rotation-invariant face detection and tracking, … | 32 | 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 |
| openai/automated-interpretability OpenAI's code and tools for automatically generating, simulating, and scoring explanations of neuron behavior in language models, based on … | 10 | 1080 | 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 |