function: deep-learning
2653 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| zju3dv/NeuralRecon NeuralRecon is a deep learning framework for real-time 3D scene reconstruction from monocular video with known camera poses. It reconstruct… | 32 | 2274 | maintenance |
| qianqianwang68/omnimotion OmniMotion is a PyTorch implementation of the ICCV 2023 paper 'Tracking Everything Everywhere All at Once', which tracks every point in a v… | 29 | 2268 | maintenance |
| andreas128/RePaint Official PyTorch implementation of RePaint, a CVPR 2022 paper on image inpainting using denoising diffusion probabilistic models. It provid… | 32 | 2267 | maintenance |
| ndrplz/ConvLSTM_pytorch A PyTorch implementation of the Convolutional LSTM (ConvLSTM) neural network module, provided as a reusable nn.Module supporting arbitrary … | 32 | 2267 | maintenance |
| mrharicot/monodepth A TensorFlow implementation of unsupervised monocular depth estimation from single images using convolutional neural networks, based on the… | 32 | 2265 | maintenance |
| crownpku/Information-Extraction-Chinese A collection of deep learning models for Chinese information extraction, including named entity recognition with IDCNN or biLSTM+CRF and re… | 32 | 2264 | maintenance |
| namisan/mt-dnn A PyTorch library implementing Multi-Task Deep Neural Networks (MT-DNN) for natural language understanding, based on Microsoft research pap… | 32 | 2257 | maintenance |
| ShoufaChen/DiffusionDet PyTorch implementation of DiffusionDet, the first diffusion-model-based object detection framework (ICCV 2023 Best Paper Finalist). It prov… | 22 | 2257 | maintenance |
| Tencent/MedicalNet MedicalNet provides a series of 3D-ResNet pre-trained models trained on 23 diverse medical imaging datasets, with PyTorch transfer-learning… | 57 | 2255 | maintenance |
| hunglc007/tensorflow-yolov4-tflite A TensorFlow 2.x implementation of YOLOv4, YOLOv4-tiny, YOLOv3, and YOLOv3-tiny object detection models, with scripts that convert original… | 32 | 2254 | maintenance |
| donnyyou/torchcv TorchCV is a PyTorch-based framework providing reimplementations of deep learning models for major computer vision tasks. It covers image c… | 32 | 2251 | maintenance |
| bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets A PyTorch implementation of several U-Net variants for image segmentation, including UNet, R2U-Net, Attention U-Net, Attention R2U-Net, and… | 32 | 2249 | maintenance |
| idealo/image-quality-assessment A Python implementation of Google's NIMA (Neural Image Assessment) models that predict the aesthetic and technical quality of images using … | 10 | 2243 | maintenance |
| Daniil-Osokin/lightweight-human-pose-estimation.pytorch A PyTorch implementation of Lightweight OpenPose for real-time 2D multi-person human pose estimation on CPU. It detects up to 18 body keypo… | 32 | 2241 | maintenance |
| microsoft/DeBERTa Official implementation of DeBERTa and DeBERTa V3, transformer-based language models with disentangled attention and decoding-enhanced BERT… | 23 | 2239 | maintenance |
| machrisaa/tensorflow-vgg A Python library implementing the VGG16 and VGG19 convolutional neural networks in TensorFlow, loading pretrained weights from npy files fo… | 32 | 2237 | maintenance |
| lonePatient/BERT-NER-Pytorch A PyTorch library for Chinese Named Entity Recognition using BERT-based models with Softmax, CRF, and Span decoding heads. It also supports… | 32 | 2235 | maintenance |
| Jongchan/attention-module Official PyTorch implementation of two attention modules for convolutional neural networks: BAM (Bottleneck Attention Module) and CBAM (Con… | 32 | 2230 | maintenance |
| gligen/GLIGEN GLIGEN is a research library for open-set grounded text-to-image generation, extending frozen diffusion models with spatial grounding via b… | 31 | 2223 | maintenance |
| KaiyangZhou/CoOp A research codebase implementing prompt learning methods (CoOp and CoCoOp) for adapting vision-language models like CLIP to downstream clas… | 32 | 2221 | maintenance |
| thu-ml/zhusuan ZhuSuan is a Python probabilistic programming library for Bayesian deep learning built on TensorFlow. It provides primitives for building p… | 32 | 2219 | maintenance |
| ProsusAI/finBERT FinBERT is a pre-trained BERT language model further trained on financial text and fine-tuned for financial sentiment classification. It is… | 32 | 2217 | maintenance |
| zuoqing1988/ZQCNN ZQCNN is a lightweight deep learning inference framework written in C/C++ that runs on Windows, Linux, and ARM-Linux. It ships with demos f… | 61 | 2214 | maintenance |
| openai/consistencydecoder A Python library from OpenAI implementing a consistency-distilled diffusion VAE decoder that improves image decoding quality for Stable Dif… | 10 | 2212 | maintenance |
| ashawkey/torch-ngp A PyTorch implementation of instant-ngp's NeRF and SDF components with CUDA extensions and an interactive GUI for training and visualizatio… | 32 | 2208 | maintenance |
| allenai/longformer Longformer is a pretrained transformer model family (including the LongformerEncoderDecoder/LED variant) that processes long documents up t… | 23 | 2205 | maintenance |
| google-research/uda Google Research's reference implementation of Unsupervised Data Augmentation (UDA), a semi-supervised learning method that uses advanced da… | 10 | 2205 | maintenance |
| yhenon/pytorch-retinanet A PyTorch implementation of the RetinaNet object detection model with focal loss, designed for readability and easy modification. It includ… | 10 | 2205 | maintenance |
| dmis-lab/biobert BioBERT is a pre-trained biomedical language representation model (based on BERT) with code for fine-tuning it on biomedical text mining ta… | 32 | 2204 | maintenance |
| lucidrains/reformer-pytorch A PyTorch implementation of the Reformer, an efficient Transformer architecture using LSH attention, reversible networks, and chunking to h… | 23 | 2191 | maintenance |
| fatchord/WaveRNN A PyTorch implementation of DeepMind's WaveRNN neural vocoder plus a Tacotron text-to-speech system, trained on LJSpeech. It supports train… | 32 | 2190 | maintenance |
| ming024/FastSpeech2 A PyTorch implementation of Microsoft's FastSpeech 2 text-to-speech model, supporting English and Mandarin with single- and multi-speaker s… | 32 | 2185 | maintenance |
| githubharald/SimpleHTR A Handwritten Text Recognition (HTR) system implemented in TensorFlow that recognizes text from images of single words or text lines, train… | 72 | 2183 | maintenance |
| tianweiy/CenterPoint Official PyTorch implementation of CenterPoint, a CVPR 2021 method that performs 3D object detection and tracking from LiDAR point clouds b… | 23 | 2182 | maintenance |
| jeonsworld/ViT-pytorch A PyTorch reimplementation of Google's Vision Transformer (ViT) for image classification, based on the paper 'An Image is Worth 16x16 Words… | 32 | 2162 | maintenance |
| facebookresearch/pycls pycls is a PyTorch-based codebase for image classification research developed by Facebook AI Research. It provides implementations of stand… | 10 | 2161 | maintenance |
| kingyiusuen/image-to-latex A PyTorch application that converts images of LaTeX math equations into LaTeX code using a ResNet-18 encoder and Transformer decoder traine… | 32 | 2159 | maintenance |
| satoshiiizuka/siggraph2016_colorization A Torch7 implementation of the SIGGRAPH 2016 paper 'Let there be Color!' that automatically colorizes grayscale images using a deep network… | 32 | 2142 | maintenance |
| nvdla/hw The hardware release of NVIDIA's open-source Deep Learning Accelerator (NVDLA), containing Verilog RTL, a C-model, testbenches, and synthes… | 32 | 2141 | maintenance |
| SeanNaren/deepspeech.pytorch A PyTorch implementation of the DeepSpeech2 speech recognition model, built on PyTorch Lightning, supporting training, testing, and inferen… | 23 | 2136 | maintenance |
| glouw/tinn Tinn is a tiny, dependency-free neural network library written in under 200 lines of C99, implementing a feedforward network with one hidde… | 32 | 2133 | maintenance |
| Mukosame/Anime2Sketch Anime2Sketch is a PyTorch-based sketch extractor that converts anime art, illustrations, and manga into line drawings using pretrained GAN … | 32 | 2128 | maintenance |
| chuanqi305/MobileNet-SSD A Caffe implementation of the MobileNet-SSD object detection network with pretrained weights on the VOC0712 dataset achieving mAP of 0.727.… | 46 | 2127 | maintenance |
| magenta/magenta-js Magenta.js is a collection of TypeScript libraries for running inference with pre-trained Magenta machine learning models directly in the b… | 72 | 2125 | maintenance |
| AIZOOTech/FaceMaskDetection An open-source face mask detection project providing a lightweight SSD-based model (1.01M parameters) with inference code for PyTorch, Tens… | 32 | 2125 | maintenance |
| atriumlts/subpixel A TensorFlow reimplementation of the efficient sub-pixel convolutional neural network (ESPCN) for single-image super-resolution, based on S… | 32 | 2123 | maintenance |
| songyouwei/ABSA-PyTorch PyTorch implementations of aspect-based sentiment analysis (ABSA) models, including attention-based and BERT-based approaches for aspect-te… | 10 | 2114 | maintenance |
| bubbliiiing/yolo3-pytorch A PyTorch implementation of the YOLOv3 object detection model with full training, prediction, and evaluation scripts. It supports training … | 23 | 2112 | maintenance |
| graphific/DeepDreamVideo A Python-based command-line tool that applies Google's DeepDream algorithm to videos by extracting frames, running a pretrained deep neural… | 32 | 2109 | maintenance |
| asappresearch/sru SRU is a PyTorch library implementing the Simple Recurrent Unit, a highly parallelizable RNN that runs 10-16x faster than cuDNN LSTM with c… | 23 | 2106 | maintenance |
| bgshih/crnn An implementation of the Convolutional Recurrent Neural Network (CRNN), combining CNN, RNN, and CTC loss for image-based sequence recogniti… | 32 | 2105 | maintenance |
| dog-qiuqiu/Yolo-Fastest Yolo-Fastest is an ultra-lightweight YOLO-based object detection algorithm and model zoo, with only ~250 MFLOPs and a 666KB ncnn model. It … | 23 | 2103 | maintenance |
| qubvel/efficientnet A Keras and TensorFlow Keras reimplementation of the EfficientNet convolutional neural network family (B0-B7), including ImageNet-pretraine… | 23 | 2100 | maintenance |
| archinetai/audio-diffusion-pytorch A PyTorch library for audio generation using diffusion models, supporting unconditional and text-conditional generation, diffusion autoenco… | 23 | 2096 | maintenance |
| facebookresearch/swav PyTorch implementation of SwAV, a self-supervised method for pre-training convolutional networks without labels by swapping cluster assignm… | 10 | 2096 | maintenance |
| koth/kcws A deep learning Chinese word segmentation toolkit implemented in C++ with TensorFlow, supporting BiLSTM+CRF and IDCNN+CRF models, plus part… | 23 | 2069 | maintenance |
| ChenglongChen/tensorflow-DeepFM A TensorFlow implementation of the DeepFM model combining factorization machines and deep neural networks for click-through-rate (CTR) pred… | 32 | 2066 | maintenance |
| tianqiraf/DouZero_For_HappyDouDiZhu A Python desktop application that applies the DouZero reinforcement-learning Dou Dizhu (Chinese card game) AI to the popular Happy DouDiZhu… | 23 | 2065 | maintenance |
| ozan-oktay/Attention-Gated-Networks A PyTorch implementation of attention gates for convolutional neural networks, applied to U-Net and VGG-16 architectures. It targets medica… | 32 | 2064 | maintenance |
| cfzd/Ultra-Fast-Lane-Detection A PyTorch implementation of the ECCV 2020 paper 'Ultra Fast Structure-aware Deep Lane Detection', providing training and evaluation code fo… | 32 | 2048 | maintenance |
| zhanghang1989/PyTorch-Encoding A PyTorch computer vision toolkit with optimized CUDA backend, providing encoding layers, synchronized multi-GPU batch normalization, and p… | 23 | 2044 | maintenance |
| alibaba/AliceMind AliceMind is Alibaba's collection of pre-trained encoder-decoder language models and related NLP techniques, including StructBERT, PALM, VE… | 23 | 2041 | maintenance |
| AnimeGAN AnimeGAN is a family of TensorFlow-based generative adversarial network models (v1, v2, v3) that convert landscape photos and videos into a… | 40 | 2036 | maintenance |
| salu133445/musegan MuseGAN is a TensorFlow-based generative adversarial network for generating polyphonic, multitrack music (bass, drums, guitar, piano, strin… | 32 | 2036 | maintenance |
| apple/ml-fastvit Official PyTorch implementation of FastViT, a fast hybrid vision transformer architecture using structural reparameterization, published at… | 28 | 2027 | maintenance |
| darglein/ADOP ADOP is a point-based differentiable neural rendering pipeline for scene refinement and novel view synthesis, implemented in C++/CUDA with … | 23 | 2026 | maintenance |
| mingyuliutw/UNIT UNIT is NVIDIA's research implementation of unsupervised image-to-image translation networks (NIPS 2017), translating images across domains… | 32 | 2025 | maintenance |
| bytedance/piano_transcription A PyTorch implementation of ByteDance's high-resolution piano transcription system that converts piano audio recordings into MIDI files. It… | 10 | 2021 | maintenance |
| tinghuiz/SfMLearner SfMLearner is a TensorFlow implementation of the CVPR 2017 paper 'Unsupervised Learning of Depth and Ego-Motion from Video'. It trains mode… | 32 | 2017 | maintenance |
| baidu/Senta Senta is Baidu's open-source sentiment analysis system built on the SKEP (Sentiment Knowledge Enhanced Pre-training) model. It provides pre… | 32 | 2015 | maintenance |
| brannondorsey/PassGAN A Python command-line implementation of the PassGAN paper that uses a Wasserstein GAN to generate password guesses, with a pretrained model… | 23 | 2012 | maintenance |
| open-mmlab/mmgeneration MMGeneration is a PyTorch-based toolkit from OpenMMLab for training and applying generative models, with a focus on GANs and support for di… | 23 | 2012 | maintenance |
| Zz-ww/SadTalker-Video-Lip-Sync A Python tool built on SadTalker that generates lip-synced video from an audio file and a source video, with configurable face/lip region e… | 30 | 2009 | maintenance |
| divelab/DIG DIG (Dive into Graphs) is a Python library built on PyTorch Geometric that provides a unified testbed for advanced graph deep learning rese… | 23 | 2007 | maintenance |
| NVlabs/alpamayo NVIDIA Alpamayo 1 is an open 10B-parameter reasoning vision-language-action (VLA) model for autonomous vehicles that pairs driving trajecto… | 59 | 2005 | maintenance |
| robbiebarrat/art-DCGAN A modified Torch implementation of DCGAN focused on generating artworks, with support for 128x128 images and resuming training from checkpo… | 32 | 2003 | maintenance |
| WongKinYiu/yolor PyTorch implementation of the YOLOR paper 'You Only Learn One Representation: Unified Network for Multiple Tasks', a real-time object detec… | 23 | 2003 | maintenance |
| haitongli/knowledge-distillation-pytorch A PyTorch framework for running knowledge distillation experiments, supporting both shallow (teacher-to-small-CNN) and deep distillation on… | 32 | 2000 | maintenance |
| JDAI-CV/FaceX-Zoo FaceX-Zoo is a PyTorch toolbox for face recognition that provides training modules with various state-of-the-art supervisory heads and back… | 32 | 1999 | maintenance |
| HobbitLong/PyContrast PyContrast is a PyTorch library providing reference implementations of contrastive learning methods such as InstDis, CMC, and MoCo, along w… | 32 | 1993 | maintenance |
| vsitzmann/siren Official PyTorch implementation of SIREN, a neural network architecture using periodic (sine) activation functions for implicit neural repr… | 32 | 1993 | maintenance |
| NVlabs/edm Official PyTorch implementation of the NeurIPS 2022 paper 'Elucidating the Design Space of Diffusion-Based Generative Models' (EDM) by NVID… | 32 | 1991 | maintenance |
| salesforce/awd-lstm-lm A PyTorch toolkit for training LSTM and QRNN language models, implementing Salesforce Research's AWD-LSTM papers. It supports word-level an… | 10 | 1989 | maintenance |
| databricks/spark-deep-learning Deep Learning Pipelines for Apache Spark, now reduced to the HorovodRunner component for distributed deep learning training via Horovod on … | 23 | 1987 | maintenance |
| ZiyaoGeng/RecLearn RecLearn is a Python/TensorFlow2.x library implementing a wide collection of recommendation algorithms, covering both matching (Top-K recom… | 23 | 1986 | maintenance |
| albertpumarola/GANimation Official PyTorch implementation of GANimation, an ECCV'18 research paper that animates facial expressions in a single image using a GAN con… | 32 | 1985 | maintenance |
| apple/ml-cvnets CVNets is Apple's open-source PyTorch library for training computer vision networks, covering classification, detection, segmentation, vide… | 32 | 1983 | maintenance |
| WuJie1010/Facial-Expression-Recognition.Pytorch A PyTorch implementation of CNN-based facial expression recognition achieving state-of-the-art accuracy on FER2013 (73.112%) and CK+ (94.64… | 32 | 1976 | maintenance |
| r9y9/deepvoice3_pytorch A PyTorch implementation of Deep Voice 3 and related convolutional neural network-based text-to-speech synthesis models. It includes traini… | 23 | 1975 | maintenance |
| SummitKwan/transparent_latent_gan TL-GAN is a Python/TensorFlow project that makes a GAN's latent space transparent by discovering feature axes, enabling controlled image sy… | 32 | 1973 | maintenance |
| JavierAntoran/Bayesian-Neural-Networks A collection of PyTorch implementations of Bayesian neural network approximate inference methods, including Bayes by Backprop, MC Dropout, … | 32 | 1969 | maintenance |
| deepseek-ai/DeepSeek-MoE DeepSeekMoE 16B is a Mixture-of-Experts language model with fine-grained expert segmentation and shared expert isolation, released with bas… | 26 | 1969 | maintenance |
| nobody132/masr MASR is an end-to-end Mandarin Chinese automatic speech recognition project built on a gated convolutional neural network (similar to Wav2L… | 23 | 1968 | maintenance |
| FoundationVision/LlamaGen LlamaGen is a family of autoregressive image generation models that apply next-token prediction, as in Llama-style LLMs, to visual generati… | 24 | 1966 | maintenance |
| ml-jku/hopfield-layers A PyTorch library implementing modern Hopfield networks with continuous states, whose update rule is equivalent to transformer attention. I… | 32 | 1959 | maintenance |
| mit-han-lab/once-for-all Once-for-All (OFA) is a PyTorch library implementing the ICLR 2020 Once-for-All network, which trains a single supernet that can be special… | 23 | 1956 | maintenance |
| hukaixuan19970627/yolov5_obb A PyTorch implementation of YOLOv5 extended for oriented (rotated) object detection using Circular Smooth Label (CSL) angle encoding. It pr… | 32 | 1947 | maintenance |
| danieldjohnson/biaxial-rnn-music-composition A Python implementation of a biaxial recurrent neural network (LSTM-based) trained to generate classical music from MIDI data. It includes … | 32 | 1926 | maintenance |
| s9xie/hed Reference implementation of Holistically-Nested Edge Detection (HED), a deep learning edge and boundary detection model built on a modified… | 23 | 1925 | maintenance |
| black0017/MedicalZooPytorch A PyTorch-based open-source library implementing state-of-the-art 3D (and some 2D) deep neural networks for multi-modal medical image segme… | 32 | 1917 | maintenance |