function: machine-learning
5378 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| 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 | 1082 | maintenance |
| Rock-100/FaceKit FaceKit is a C++ toolkit implementing Progressive Calibration Networks (PCN) for real-time rotation-invariant face detection and tracking, … | 32 | 1082 | 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 | 1082 | 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 |
| akelleh/causality A Python library providing tools for causal inference and analysis from observational data, built as an extension of pandas DataFrames. It … | 27 | 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 | 1081 | maintenance |
| weiyithu/SurroundOcc SurroundOcc is the official PyTorch implementation of an ICCV 2023 paper predicting dense 3D volumetric occupancy from multi-camera images … | 43 | 1080 | maintenance |
| rhiever/datacleaner A Python package and command-line tool that automatically cleans pandas DataFrames, handling missing values and encoding categorical variab… | 23 | 1079 | 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 | 1079 | maintenance |
| locuslab/mpc.pytorch mpc.pytorch is a fast and differentiable model predictive control (MPC) solver implemented as a PyTorch library. It solves receding-horizon… | 32 | 1078 | 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 |
| 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 | 1075 | 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 | 1075 | maintenance |
| sunset1995/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 | 1074 | 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 | 1072 | maintenance |
| Wangt-CN/DisCo DisCo is a CVPR 2024 research codebase for referring human dance generation, producing realistic dance images and videos from a reference h… | 29 | 1072 | maintenance |
| ayoolaolafenwa/PixelLib PixelLib is a Python library for performing semantic and instance segmentation of objects in images and videos with just a few lines of cod… | 23 | 1072 | maintenance |
| huihut/OpenCV-MinGW-Build A repository of prebuilt OpenCV binaries (32-bit and 64-bit) compiled with MinGW on Windows, covering versions from 3.3.1 through 4.5.5 inc… | 23 | 1072 | 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… | 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 |
| SergeyMorugin/ostagram Ostagram is a self-hosted Ruby on Rails web application that applies neural style transfer to user-uploaded photos, blending them with the … | 50 | 1068 | 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 |
| 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 | 1068 | maintenance |
| Li-Chongyi/Zero-DCE PyTorch implementation of Zero-DCE (Zero-Reference Deep Curve Estimation), a CVPR 2020 model that enhances low-light images without paired … | 32 | 1067 | maintenance |
| EdVince/Stable-Diffusion-NCNN A C++ implementation of Stable Diffusion using the NCNN inference framework, supporting both txt2img and img2img. It runs on x86 Windows ex… | 23 | 1067 | 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 | 1066 | maintenance |
| omerbt/MultiDiffusion Official PyTorch implementation of MultiDiffusion (ICML 2023), a training-free framework that fuses multiple diffusion paths over a pre-tra… | 31 | 1066 | 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 | 1066 | maintenance |
| PyPatel/Options-Trading-Strategies-in-Python A collection of Python scripts implementing options trading strategies based on technical indicators such as VIX, Put/Call Ratio, TRIN, and… | 32 | 1064 | maintenance |
| facebookresearch/mbrl-lib MBRL-Lib is a PyTorch-based toolbox from Facebook Research for developing model-based reinforcement learning algorithms. It provides interc… | 10 | 1064 | maintenance |
| JiehangXie/PaddleBoBo PaddleBoBo is a Python project built on PaddlePaddle (with PaddleSpeech and PaddleGAN) that quickly generates a virtual streamer (VTuber) f… | 32 | 1063 | 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 | 1063 | 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 |
| J535D165/recordlinkage A modular Python toolkit for record linkage and duplicate detection across datasets, built on pandas and numpy. It provides indexing (block… | 23 | 1062 | maintenance |
| open-mmlab/mmflow MMFlow is an open-source optical flow toolbox and benchmark built on PyTorch as part of the OpenMMLab project. It provides a unified, modul… | 23 | 1061 | 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 | 1061 | maintenance |
| jsyoon0823/TimeGAN Reference implementation of TimeGAN, a generative adversarial network framework for generating synthetic time-series data, published at Neu… | 62 | 1060 | maintenance |
| zhaoweicai/cascade-rcnn A C++/Caffe implementation of Cascade R-CNN and other popular two-stage object detection frameworks such as Faster R-CNN, R-FCN, and FPN. I… | 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… | 29 | 1060 | maintenance |
| featurecat/lizzie Lizzie is a Java-based graphical interface for analyzing Go (baduk) games in real time using the Leela Zero engine. It displays win rates, … | 23 | 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… | 29 | 1059 | maintenance |
| BurguerJohn/Dain-App Dain-App is a video frame interpolation application built on the DAIN (Depth-Aware Video Frame Interpolation) deep learning model. It provi… | 23 | 1059 | maintenance |
| SeetaFace SeetaFace is an open-source, full-stack face recognition toolkit written in standard C++ with no third-party dependencies. It provides face… | 32 | 1058 | maintenance |
| chenyuntc/PyTorchText A PyTorch implementation of multiple text classification models (TextCNN, TextRNN/LSTM, RCNN, FastText, inception CNN) that won 1st place i… | 32 | 1056 | 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 | 1056 | maintenance |
| piergiaj/pytorch-i3d A PyTorch port of DeepMind's I3D (Inflated 3D ConvNet) models pretrained on the Kinetics dataset for video action recognition. It includes … | 32 | 1056 | maintenance |
| matghazaryan/SOMotionDetector An Objective-C library for iOS that detects user motion type (not moving, walking, running, automotive) and counts steps using CoreLocation… | 23 | 1056 | maintenance |
| HRNet/HRNet-Image-Classification Official PyTorch implementation and training code for HRNet (High-Resolution Network) image classification models on ImageNet. It provides … | 23 | 1056 | maintenance |
| huggingface/optimum-quanto Optimum Quanto is a PyTorch quantization backend for Hugging Face Optimum that quantizes model weights (int2/int4/int8/float8) and activati… | 69 | 1054 | maintenance |
| raminmh/CfC Reference implementations of Closed-form Continuous-time (CfC) neural networks, a fast closed-form approximation of liquid time-constant ne… | 23 | 1054 | maintenance |
| keras-team/keras-cv KerasCV is a library of modular computer vision components built on Keras 3 that work natively with TensorFlow, JAX, or PyTorch. It provide… | 10 | 1054 | maintenance |
| aliutkus/speechmetrics A Python library that wraps several objective speech quality metrics (MOSNet, BSSEval, STOI, PESQ, SRMR, SISDR) behind a unified API. It su… | 32 | 1053 | maintenance |
| Edresson/YourTTS YourTTS is a zero-shot multi-speaker text-to-speech and voice conversion model built on VITS, implemented in the Coqui TTS framework. It su… | 23 | 1053 | maintenance |
| microsoft/Oscar Oscar is Microsoft's research code for object-semantics aligned cross-modal pre-training of vision-language models, with VinVL providing im… | 10 | 1053 | maintenance |
| ashkamath/mdetr MDETR (Modulated Detection) is a PyTorch research codebase for end-to-end multi-modal object detection that grounds free-form text queries … | 32 | 1052 | maintenance |
| damo-cv/TransReID Official PyTorch implementation of TransReID, an ICCV 2021 paper applying vision transformers to object re-identification. It provides trai… | 32 | 1052 | maintenance |
| lhelontra/tensorflow-on-arm A set of shell scripts and Docker-based build tooling for cross-compiling TensorFlow wheels for ARM boards like Raspberry Pi and Odroid. It… | 23 | 1052 | maintenance |
| microsoft/Cognitive-Samples-IntelligentKiosk A UWP sample application from Microsoft showcasing hands-free kiosk-style demos built on Azure Cognitive Services (Face, Computer Vision, T… | 10 | 1052 | maintenance |
| caoscott/SReC SReC is the official PyTorch implementation of the paper 'Lossless Image Compression through Super-Resolution', which frames lossless image… | 32 | 1050 | maintenance |
| isaac-sim/OmniIsaacGymEnvs A collection of GPU-accelerated reinforcement learning environments built on NVIDIA Isaac Sim and the omni.isaac.gym framework, with PPO tr… | 10 | 1049 | maintenance |
| yangxudong/deeplearning A collection of deep learning model training, evaluation, and prediction code implemented with TensorFlow's high-level Estimator API, with … | 32 | 1048 | maintenance |
| microsoft/SimMIM Official PyTorch implementation of SimMIM, a simple framework for masked image modeling (self-supervised visual pre-training) from Microsof… | 32 | 1048 | maintenance |
| uber/fiber Fiber is a Python distributed computing library for modern computer clusters that exposes the same API as Python's standard multiprocessing… | 10 | 1047 | maintenance |
| yuval-alaluf/restyle-encoder Official PyTorch implementation of ReStyle, a residual-based StyleGAN encoder that inverts real images into GAN latent codes via iterative … | 32 | 1046 | maintenance |
| thunlp/OpenDelta OpenDelta is a Python library for parameter-efficient tuning (delta tuning) of pretrained language models, letting users attach small train… | 23 | 1046 | maintenance |
| 4uiiurz1/pytorch-nested-unet A PyTorch implementation of the UNet++ (Nested U-Net) architecture for image segmentation, based on the paper 'UNet++: A Nested U-Net Archi… | 32 | 1045 | maintenance |
| pkhungurn/talking-head-anime-3-demo Demo programs for the Talking Head(?) Anime 3 project, which animates an anime character from a single image using machine learning. It inc… | 32 | 1045 | maintenance |
| YuwenXiong/py-R-FCN A Python implementation of R-FCN (Region-based Fully Convolutional Networks) for object detection, modified from the official MATLAB code a… | 32 | 1043 | maintenance |
| facebookresearch/FixRes FixRes is a PyTorch implementation of the NeurIPS 2019 paper 'Fixing the train-test resolution discrepancy', providing training and fine-tu… | 10 | 1043 | maintenance |
| haltakov/natural-language-image-search A demo application that searches Unsplash photos using natural language descriptions, powered by OpenAI's CLIP model over the ~2M photo Uns… | 23 | 1042 | maintenance |
| pixray/pixray Pixray is a Python library and command-line utility for text-to-image generation, combining CLIP-guided GAN imagery, pixel-art drawers, and… | 23 | 1042 | maintenance |
| asingh33/CNNGestureRecognizer A desktop application that recognizes hand gestures from webcam video using a convolutional neural network built with Keras, TensorFlow/The… | 60 | 1041 | maintenance |
| IBM/MAX-Image-Resolution-Enhancer An IBM Model Asset Exchange project that deploys an SRGAN-based image super-resolution model as a web service in a Docker container. It ups… | 41 | 1041 | maintenance |
| thuml/Anomaly-Transformer Official PyTorch implementation of the Anomaly Transformer model from the ICLR 2022 Spotlight paper on unsupervised time series anomaly det… | 32 | 1041 | maintenance |
| SysCV/sam-pt SAM-PT extends the Segment Anything Model to zero-shot video segmentation by combining SAM with sparse point-based tracking (PIPS, CoTracke… | 29 | 1041 | maintenance |
| keijiro/Pix2Pix A Unity library that runs pix2pix image-to-image translation neural networks in real time using compute shaders. It includes its own infere… | 23 | 1041 | maintenance |
| descriptinc/melgan-neurips Official PyTorch implementation of MelGAN, a GAN-based non-autoregressive vocoder that inverts mel-spectrograms into raw audio waveforms fo… | 32 | 1040 | maintenance |
| MaybeShewill-CV/CRNN_Tensorflow A TensorFlow implementation of CRNN (CNN + Bi-LSTM + CTC loss) for scene text recognition, based on the Shi et al. paper. It includes pretr… | 32 | 1039 | maintenance |
| Shunichi09/PythonLinearNonlinearControl A Python library implementing linear and nonlinear control theory algorithms such as MPC, iLQR, DDP, MPPI, and CEM using only NumPy and Sci… | 23 | 1039 | maintenance |
| huggingface/pytorch-pretrained-BigGAN A PyTorch reimplementation of DeepMind's BigGAN generator with pretrained weights at 128, 256, and 512 pixel resolutions, plus scripts to c… | 23 | 1039 | maintenance |
| LionelJouin/PiP-Tool A Windows desktop application that enables Picture-in-Picture mode by capturing a selected region of any window and displaying it in an alw… | 23 | 1039 | maintenance |
| google-research/morph-net MorphNet is a TensorFlow library for resource-constrained learning of deep network structure. It adds regularizers during training that ind… | 72 | 1038 | maintenance |
| JIA-Lab-research/SNR-Aware-Low-Light-Enhance Official PyTorch implementation of the CVPR 2022 paper 'SNR-aware Low-Light Image Enhancement'. It combines SNR-aware transformers and conv… | 32 | 1038 | maintenance |
| CalciferZh/minimal-hand A Python library for real-time 3D hand motion capture from a single RGB camera, running at over 100fps. It combines a deep network (DetNet)… | 23 | 1038 | maintenance |
| yashbhalgat/HashNeRF-pytorch A pure PyTorch implementation of NVIDIA's Instant-NGP multi-resolution hash encoding for training NeRFs, built on top of NeRF-pytorch. It e… | 32 | 1037 | maintenance |
| google-research/deeplab2 DeepLab2 is a TensorFlow library from Google Research providing a unified, state-of-the-art codebase for dense pixel labeling tasks such as… | 10 | 1037 | maintenance |
| Xwin-LM/Xwin-LM Xwin-LM is an open-source project for LLM alignment technologies including supervised fine-tuning, reward models, reject sampling, and RLHF… | 28 | 1036 | maintenance |
| sicara/tf-explain tf-explain is a Python library implementing interpretability methods (Grad CAM, occlusion sensitivity, vanilla gradients, etc.) for tf.kera… | 23 | 1036 | maintenance |
| pesser/stable-diffusion The development repository for Stable Diffusion and Latent Diffusion Models, containing research code, training scripts, and pretrained mod… | 32 | 1033 | maintenance |
| goberoi/faceit A Python script that simplifies swapping faces in videos using the deepfakes/faceswap library, with training data sourced from YouTube vide… | 32 | 1032 | maintenance |
| ClementPinard/SfmLearner-Pytorch A PyTorch reimplementation of SfMLearner (Zhou et al., CVPR 2017), which jointly learns monocular depth estimation and camera ego-motion fr… | 66 | 1031 | maintenance |
| xingyizhou/ExtremeNet Official PyTorch implementation of ExtremeNet, a CVPR 2019 bottom-up object detection method that detects four extreme points and one cente… | 32 | 1031 | maintenance |
| vincentherrmann/pytorch-wavenet A PyTorch implementation of the WaveNet neural audio generation architecture with fast generation support. It includes dataset creation fro… | 32 | 1031 | maintenance |
| ArrowLuo/CLIP4Clip Official PyTorch implementation of the CLIP4Clip paper, a video-text retrieval model that transfers CLIP knowledge to end-to-end video clip… | 23 | 1031 | maintenance |
| kakaobrain/rq-vae-transformer The official PyTorch implementation of 'Autoregressive Image Generation using Residual Quantization' (CVPR 2022), implementing RQ-VAE and R… | 32 | 1030 | maintenance |
| graphdeeplearning/graphtransformer Reference implementation of the Graph Transformer architecture from the paper 'A Generalization of Transformer Networks to Graphs' (DLG-AAA… | 32 | 1030 | maintenance |
| qubvel/ttach TTAch is a Python library for image test time augmentation (TTA) with PyTorch. It wraps existing models to apply augmentations like flips, … | 23 | 1030 | maintenance |