function: deep-learning
2653 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| datitran/face2face-demo A pix2pix demo that learns from facial landmarks and translates them into a trained target face, with a real-time webcam application. It in… | 32 | 1461 | maintenance |
| facebookresearch/MaskFormer MaskFormer is a PyTorch/Detectron2-based implementation of the NeurIPS 2021 paper 'Per-Pixel Classification is Not All You Need for Semanti… | 10 | 1460 | maintenance |
| una-dinosauria/3d-pose-baseline A TensorFlow implementation of a simple yet effective baseline for 3D human pose estimation from 2D keypoints, published at ICCV 2017. It i… | 32 | 1459 | maintenance |
| DLTK/DLTK DLTK is a Python neural networks toolkit built on TensorFlow for deep learning on medical images. It provides reference model implementatio… | 32 | 1458 | maintenance |
| microsoft/NeuronBlocks NeuronBlocks is an NLP deep learning modeling toolkit from Microsoft that lets users build end-to-end neural network training and inference… | 23 | 1452 | maintenance |
| HuwCampbell/grenade Grenade is a composable, dependently typed deep learning library for Haskell that lets users specify neural network architectures at the ty… | 23 | 1449 | maintenance |
| hysts/pytorch_image_classification A PyTorch library implementing many image classification architectures (ResNet, DenseNet, WRN, PyramidNet, SENet, etc.) and augmentation te… | 10 | 1449 | maintenance |
| BigScience BigScience is a research project training large transformer language models (BERT, GPT-style) at scale, built on a fork of Megatron-LM inte… | 32 | 1448 | maintenance |
| bojone/attention A small Python library providing implementations of the attention mechanism from 'Attention is All You Need' for TensorFlow/Keras. It is no… | 32 | 1448 | maintenance |
| mit-han-lab/proxylessnas ProxylessNAS is a neural architecture search (NAS) framework that directly searches CNN architectures on the target task and target hardwar… | 32 | 1447 | maintenance |
| megvii-research/ML-GCN A PyTorch implementation of ML-GCN, the CVPR 2019 paper 'Multi-Label Image Recognition with Graph Convolutional Networks'. It provides trai… | 32 | 1446 | maintenance |
| ZPdesu/Barbershop Official PyTorch implementation of Barbershop (SIGGRAPH Asia 2021), a GAN-inversion-based method for compositing images using segmentation … | 32 | 1439 | maintenance |
| liyaguang/DCRNN A TensorFlow implementation of the Diffusion Convolutional Recurrent Neural Network (DCRNN) from the ICLR 2018 paper on data-driven traffic… | 32 | 1439 | maintenance |
| mchong6/JoJoGAN JoJoGAN is the official PyTorch implementation of a one-shot face stylization method that finetunes a pretrained StyleGAN using GAN inversi… | 32 | 1438 | maintenance |
| AshwinRJ/Federated-Learning-PyTorch A PyTorch implementation of the FedAvg algorithm from the paper 'Communication-Efficient Learning of Deep Networks from Decentralized Data'… | 32 | 1437 | maintenance |
| rail-berkeley/softlearning Softlearning is a deep reinforcement learning toolbox for training maximum entropy policies in continuous domains, and the official impleme… | 32 | 1437 | maintenance |
| xuebinqin/BASNet BASNet is the official PyTorch implementation of the CVPR 2019 paper 'BASNet: Boundary-Aware Salient Object Detection', a deep learning mod… | 32 | 1436 | maintenance |
| yangyanli/PointCNN PointCNN is a deep learning framework for feature learning from 3D point clouds, applying convolution on X-transformed points to handle the… | 55 | 1434 | maintenance |
| pytorch/functorch functorch is a library providing JAX-like composable function transforms (vmap, grad, vjp, jvp) for PyTorch, enabling efficient per-sample … | 10 | 1434 | maintenance |
| denizyuret/Knet.jl Knet (Koç University deep learning framework) is a deep learning library implemented in Julia with GPU support and automatic differentiatio… | 23 | 1432 | maintenance |
| SamLynnEvans/Transformer A PyTorch implementation of the Transformer seq2seq model designed to build language translators from parallel corpora. It accompanies a tu… | 32 | 1430 | maintenance |
| vlfeat/matconvnet MatConvNet is a MATLAB toolbox implementing convolutional neural networks (CNNs) for computer vision applications. It supports training and… | 32 | 1430 | maintenance |
| qfgaohao/pytorch-ssd A PyTorch implementation of the SSD (Single Shot MultiBox Detector) object detection algorithm with MobileNetV1, MobileNetV2, and VGG backb… | 32 | 1429 | maintenance |
| OpenLMLab/MOSS-RLHF MOSS-RLHF is the open-source companion code for the paper 'Secrets of RLHF in Large Language Models Part I: PPO', providing implementations… | 29 | 1429 | maintenance |
| KaiyangZhou/Dassl.pytorch Dassl is a PyTorch toolbox for research in domain adaptation, domain generalization, and semi-supervised learning. It provides modular comp… | 32 | 1428 | maintenance |
| theAIGuysCode/yolov4-deepsort A Python implementation of multi-object tracking that combines YOLOv4 object detection with the Deep SORT tracking algorithm using TensorFl… | 32 | 1428 | maintenance |
| facebookresearch/diplomacy_cicero Research code and model checkpoints for Cicero and Diplodocus, AI agents that play the board game Diplomacy at human level by combining lan… | 10 | 1428 | maintenance |
| open-mmlab/mmhuman3d MMHuman3D is an open-source PyTorch-based toolbox and benchmark for 3D human parametric models (e.g., SMPL, SMPL-X) in computer vision and … | 23 | 1426 | maintenance |
| torchgan/torchgan TorchGAN is a PyTorch-based research framework for designing and training Generative Adversarial Networks. It provides modular building blo… | 23 | 1425 | maintenance |
| google-research/disentanglement_lib disentanglement_lib is an open-source Python library for research on learning disentangled representations, supporting models like BetaVAE,… | 10 | 1425 | maintenance |
| rmokady/CLIP_prefix_caption Official implementation of ClipCap, a CLIP-based image captioning model that maps CLIP image encodings to a GPT-2 prefix to generate captio… | 32 | 1423 | maintenance |
| GOATmessi8/RFBNet PyTorch implementation of RFBNet, a Receptive Field Block Net object detector presented at ECCV 2018. It enhances SSD-style detectors with … | 32 | 1419 | maintenance |
| BangguWu/ECANet Official PyTorch implementation of ECA-Net, an efficient channel attention module for deep convolutional neural networks published at CVPR … | 32 | 1416 | maintenance |
| cdpierse/transformers-interpret A Python library providing model explainability for Hugging Face Transformers models, built on Captum. It explains text classification, que… | 23 | 1416 | maintenance |
| bermanmaxim/LovaszSoftmax Standalone PyTorch and TensorFlow implementations of the Lovász-Softmax and Lovász Hinge loss functions for optimizing the Jaccard index (I… | 32 | 1410 | maintenance |
| chrischoy/3D-R2N2 3D-R2N2 is a PyTorch-based implementation of a recurrent neural network that reconstructs voxelized 3D models of objects from one or multip… | 32 | 1410 | maintenance |
| yu-changqian/TorchSeg A fast, modular PyTorch reference implementation for training and evaluating semantic segmentation models such as FCN, DFN, BiSeNet, PSPNet… | 23 | 1410 | maintenance |
| tonylins/pytorch-mobilenet-v2 A PyTorch implementation of the MobileNetV2 architecture with automatic download of ImageNet pretrained weights. It reproduces the official… | 32 | 1406 | maintenance |
| facebookresearch/end-to-end-negotiator A PyTorch research codebase from Facebook AI Research implementing end-to-end learning for negotiation dialogues, including models for hier… | 10 | 1400 | maintenance |
| gotzmann/llama.go A pure-Go reimplementation of llama.cpp-style LLM inference, running LLaMA-family models on CPU without C++ dependencies. It includes tenso… | 21 | 1397 | maintenance |
| opendilab/DI-star DI-star is a large-scale distributed training platform for building StarCraft II game AI, including supervised and reinforcement learning t… | 37 | 1393 | maintenance |
| swz30/MPRNet MPRNet is a PyTorch implementation of a multi-stage progressive image restoration network published at CVPR 2021. It provides pretrained mo… | 32 | 1393 | maintenance |
| yao8839836/text_gcn The official implementation of Text GCN, a Graph Convolutional Network approach for text classification published at AAAI 2019. It builds w… | 32 | 1389 | maintenance |
| yinboc/liif LIIF is the official PyTorch implementation of the CVPR 2021 paper 'Learning Continuous Image Representation with Local Implicit Image Func… | 32 | 1388 | maintenance |
| 198808xc/Pangu-Weather Official implementation of Pangu-Weather, a deep learning model for fast, accurate medium-range global weather forecasting using 3D neural … | 22 | 1388 | maintenance |
| doonny/PipeCNN PipeCNN is an OpenCL-based FPGA accelerator for large-scale convolutional neural network inference, written in C with pipelined kernels. It… | 32 | 1386 | maintenance |
| chrisdonahue/wavegan Official TensorFlow implementation of WaveGAN, a generative adversarial network that learns to synthesize raw audio waveforms from examples… | 32 | 1383 | maintenance |
| wouterkool/attention-learn-to-route A PyTorch implementation of the attention-based neural model from the ICLR 2019 paper 'Attention, Learn to Solve Routing Problems!', traine… | 32 | 1383 | maintenance |
| bytedance/music_source_separation A PyTorch implementation of music source separation that splits songs into sources like vocals and accompaniment. It ships pretrained check… | 10 | 1383 | maintenance |
| DeepGraphLearning/KnowledgeGraphEmbedding A PyTorch toolkit implementing the RotatE knowledge graph embedding model along with TransE, ComplEx, DistMult, and pRotatE. It trains and … | 32 | 1381 | maintenance |
| PSPNet Reference implementation of the Pyramid Scene Parsing Network (PSPNet), a CVPR 2017 semantic segmentation model that won the ImageNet Scene… | 32 | 1378 | maintenance |
| msracver/Deep-Image-Analogy Official C++/CUDA implementation of the SIGGRAPH 2017 'Visual Attribute Transfer through Deep Image Analogy' technique from Microsoft Resea… | 32 | 1370 | maintenance |
| zalandoresearch/pytorch-ts PyTorchTS is a PyTorch-based probabilistic time series forecasting library that uses GluonTS as its backend API for loading, transforming, … | 23 | 1370 | maintenance |
| nv-tlabs/lift-splat-shoot PyTorch implementation of Lift-Splat-Shoot (ECCV 2020), an end-to-end model that converts images from arbitrary multi-camera rigs into a bi… | 32 | 1369 | maintenance |
| open-mmlab/mmfashion MMFashion is an open-source PyTorch-based toolbox for visual fashion analysis from the OpenMMLab project. It provides modular implementatio… | 32 | 1369 | maintenance |
| akamaster/pytorch_resnet_cifar10 A PyTorch implementation of ResNet architectures (ResNet20 through ResNet1202) for CIFAR10/CIFAR100 that faithfully matches the original pa… | 32 | 1368 | maintenance |
| Cysu/open-reid Open-ReID is a lightweight Python library for person re-identification built on PyTorch, providing a uniform dataset interface, models, and… | 32 | 1368 | maintenance |
| mdeff/cnn_graph Reference implementation of the NIPS 2016 paper 'Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering' (ChebNet),… | 32 | 1366 | maintenance |
| JonasGeiping/cramming A research framework for pretraining BERT-type language models from scratch on a single consumer GPU in one day, replicating the paper 'Cra… | 22 | 1366 | maintenance |
| wzzheng/TPVFormer TPVFormer is a CVPR 2023 research implementation of a tri-perspective view transformer for vision-based 3D semantic occupancy prediction, s… | 32 | 1364 | maintenance |
| ali-vilab/ACE_plus ACE++ is a Python library and model release from Alibaba's Tongyi Lab for instruction-based image creation and editing via context-aware co… | 28 | 1364 | maintenance |
| taoxugit/AttnGAN A PyTorch implementation of AttnGAN, a fine-grained text-to-image generative adversarial network with multi-stage attention mechanisms from… | 32 | 1363 | maintenance |
| sail-sg/poolformer PoolFormer is a PyTorch implementation of the CVPR 2022 paper 'MetaFormer Is Actually What You Need for Vision', which replaces attention i… | 23 | 1362 | maintenance |
| MoyGcc/vid2avatar Vid2Avatar is the official PyTorch implementation of a CVPR 2023 method that reconstructs detailed 3D human avatars from monocular in-the-w… | 56 | 1360 | maintenance |
| atulapra/Emotion-detection A Python application that performs real-time facial emotion detection from a webcam feed using a CNN trained on the FER-2013 dataset. It us… | 32 | 1359 | maintenance |
| mayuelala/FollowYourPose Official PyTorch implementation of Follow-Your-Pose (AAAI 2024), a pose-guided text-to-video generation model that tunes a text-to-image mo… | 21 | 1358 | maintenance |
| nvidia-cosmos/cosmos-predict2.5 NVIDIA Cosmos-Predict2.5 is a family of world foundation models (WFMs) that generate video predictions of future world states for physical … | 72 | 1355 | maintenance |
| orobix/retina-unet A Python implementation of a U-Net convolutional neural network for segmenting blood vessels in retina fundus images. It performs binary pi… | 32 | 1354 | maintenance |
| dorarad/gansformer GANformer is a research implementation of a generative adversarial transformer that uses a bipartite attention structure for efficient high… | 23 | 1344 | maintenance |
| PeizeSun/SparseR-CNN Sparse R-CNN is a PyTorch implementation (built on Detectron2) of the CVPR 2021 / PAMI 2023 paper 'End-to-End Object Detection with Learnab… | 23 | 1343 | maintenance |
| HobbitLong/CMC Official PyTorch implementation of Contrastive Multiview Coding (CMC), a self-supervised visual representation learning method that contras… | 32 | 1340 | maintenance |
| bloc97/CrossAttentionControl An unofficial Jupyter Notebook implementation of the 'Prompt-to-Prompt Image Editing with Cross Attention Control' paper adapted for Stable… | 32 | 1337 | maintenance |
| torrvision/crfasrnn Reference implementation of CRF-RNN, an ICCV 2015 semantic image segmentation method that integrates conditional random fields into a neura… | 23 | 1335 | maintenance |
| maudzung/Complex-YOLOv4-Pytorch A PyTorch implementation of Complex-YOLO, a YOLOv4-based model for real-time 3D object detection on LiDAR point clouds. It supports distrib… | 32 | 1327 | maintenance |
| google/sg2im A PyTorch research implementation of the CVPR 2018 paper 'Image Generation from Scene Graphs' by Johnson et al. It converts a structured sc… | 10 | 1325 | maintenance |
| CharlesShang/DCNv2 A PyTorch implementation of Deformable Convolutional Networks v2 (DCNv2), providing CUDA/CPU operators for deformable convolution as a drop… | 71 | 1323 | maintenance |
| facebookresearch/moco-v3 A PyTorch implementation of MoCo v3, a self-supervised contrastive learning method for ResNet and Vision Transformer (ViT) models. It inclu… | 10 | 1323 | maintenance |
| CSTR-Edinburgh/merlin Merlin is a toolkit from the University of Edinburgh's CSTR for building deep neural network models for statistical parametric speech synth… | 23 | 1321 | maintenance |
| aitorzip/PyTorch-CycleGAN A clean, readable PyTorch implementation of CycleGAN for unpaired image-to-image translation using cycle-consistent adversarial networks. I… | 32 | 1319 | maintenance |
| hukkelas/DeepPrivacy DeepPrivacy is a PyTorch-based GAN that automatically anonymizes faces in images and videos by generating realistic synthetic replacements.… | 32 | 1316 | maintenance |
| msracver/Deep-Feature-Flow Official MXNet implementation of Deep Feature Flow (CVPR 2017), an end-to-end framework for video recognition such as object detection and … | 32 | 1315 | maintenance |
| szagoruyko/wide-residual-networks Reference implementation of Wide Residual Networks (WRN), a ResNet variant that trades depth for width to train faster and achieve state-of… | 32 | 1315 | maintenance |
| facebookresearch/svoice SVoice is a PyTorch implementation of the ICML paper 'Voice Separation with an Unknown Number of Multiple Speakers' from Facebook AI Resear… | 10 | 1315 | maintenance |
| LinXueyuanStdio/LaTeX_OCR_PRO A deep-learning application that converts images of math formulas (printed, handwritten, and Chinese-mixed) into LaTeX code using a Seq2Seq… | 32 | 1310 | maintenance |
| d-li14/involution Official PyTorch implementation of the involution neural operator from the CVPR 2021 paper 'Involution: Inverting the Inherence of Convolut… | 32 | 1310 | maintenance |
| tysam-code/hlb-CIFAR10 A single-file PyTorch implementation that trains a neural network to 94% accuracy on CIFAR-10 in under 6.3 seconds on a single A100 GPU, fo… | 22 | 1310 | maintenance |
| NVlabs/prismer Official PyTorch implementation of Prismer, a data- and parameter-efficient vision-language model that ensembles pre-trained task-specific … | 30 | 1309 | maintenance |
| mit-han-lab/data-efficient-gans Official implementation of Differentiable Augmentation (DiffAugment), a NeurIPS 2020 method that improves GAN training data efficiency by a… | 32 | 1308 | maintenance |
| EricGuo5513/momask-codes Official PyTorch implementation of MoMask (CVPR 2024), a masked modeling framework that generates 3D human motion sequences from text promp… | 27 | 1308 | maintenance |
| chickenbestlover/RNN-Time-series-Anomaly-Detection A PyTorch implementation of an RNN-based time-series anomaly detector using a two-stage strategy of multi-step prediction followed by anoma… | 32 | 1305 | maintenance |
| senlinuc/caffe_ocr An experimental research project built on Caffe implementing CNN+BLSTM+CTC text recognition architectures, with modifications for LSTM, war… | 32 | 1305 | maintenance |
| jannerm/diffuser Official code for the ICML 2022 paper 'Planning with Diffusion for Flexible Behavior Synthesis', which plans by denoising trajectories with… | 32 | 1304 | maintenance |
| nlpyang/PreSumm PreSumm is the official PyTorch implementation of the EMNLP 2019 paper 'Text Summarization with Pretrained Encoders', providing BERT-based … | 32 | 1303 | maintenance |
| digantamisra98/Mish Official repository for the Mish activation function, a self-regularized non-monotonic neural activation function published at BMVC 2020. I… | 74 | 1299 | maintenance |
| foolwood/DaSiamRPN PyTorch implementation of DaSiamRPN, an ECCV 2018 distractor-aware Siamese network for visual object tracking, winner of the VOT-18 real-ti… | 32 | 1299 | maintenance |
| vietnh1009/Super-mario-bros-PPO-pytorch A PyTorch implementation of the Proximal Policy Optimization (PPO) algorithm that trains an agent to play Super Mario Bros via OpenAI Gym. … | 32 | 1299 | maintenance |
| NVlabs/DG-Net DG-Net is a PyTorch implementation of the CVPR 2019 (Oral) paper 'Joint Discriminative and Generative Learning for Person Re-identification… | 32 | 1298 | maintenance |
| Netflix/vectorflow Vectorflow is a minimalist neural network library written in D, optimized for sparse data and single-machine training environments. It is d… | 32 | 1294 | maintenance |
| google-research/multilingual-t5 Code and resources for mT5, a massively multilingual text-to-text transformer pretrained on the mC4 corpus covering 101 languages. It repro… | 10 | 1294 | maintenance |
| bubbliiiing/deeplabv3-plus-pytorch A PyTorch implementation of the DeepLabv3+ semantic segmentation model with MobileNetV2 and Xception backbones. It includes scripts for tra… | 23 | 1292 | maintenance |