Ross ROSS = Recommend OSS · open-source software intelligence for agents

function: deep-learning

2653 products, primary matches first, then adoption-weighted; health v2 shown.

ProductHealth v2StarsMaturity
spring-media/TransformerTTS
A TensorFlow 2 implementation of a non-autoregressive Transformer-based neural network for text-to-speech synthesis, based on FastSpeech an…
101161maintenance
cheungdaven/DeepRec
DeepRec is an open-source Python/TensorFlow toolkit implementing deep learning based recommendation models for rating prediction, top-N ite…
321160maintenance
wyharveychen/CloserLookFewShot
Reference PyTorch implementation of the ICLR 2019 paper 'A Closer Look at Few-shot Classification', providing a testbed for empirical evalu…
321159maintenance
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…
321156maintenance
abhishekkrthakur/tez
Tez is a lightweight, simple trainer library for PyTorch that simplifies training loops while keeping users close to native PyTorch. It sup…
231155maintenance
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…
321154maintenance
houqb/CoordAttention
A PyTorch implementation of Coordinate Attention (CoordAttention), an efficient attention mechanism for mobile networks introduced in a CVP…
321152maintenance
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…
321150maintenance
bubbliiiing/yolov5-pytorch
A PyTorch implementation of the YOLOv5 (v5.0) object detection model with full training, prediction, and evaluation pipelines. It is design…
231149maintenance
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…
461148maintenance
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…
321148maintenance
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…
231148maintenance
tensorflow/benchmarks
A benchmark framework for TensorFlow, containing PerfZero for running and reporting TensorFlow benchmarks and legacy CNN benchmark scripts …
101148maintenance
joe-siyuan-qiao/DetectoRS
Official PyTorch implementation of DetectoRS, a state-of-the-art object detection and instance segmentation model using Recursive Feature P…
321147maintenance
tech-srl/code2vec
Official TensorFlow implementation of the code2vec model from the POPL'2019 paper, which learns distributed vector representations of code …
321147maintenance
ZhuiyiTechnology/roformer
RoFormer is an MLM pre-trained language model built on rotary position embeddings (RoPE), a relative position encoding method with strong t…
321146maintenance
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…
311146maintenance
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…
321141maintenance
google-research/mixmatch
Reference implementation of MixMatch, a holistic semi-supervised learning algorithm from a Google Research paper. It includes dataset prepa…
101139maintenance
HRNet/HRNet-Facial-Landmark-Detection
Official PyTorch implementation of HRNet-based facial landmark detection from the TPAMI paper 'Deep High-Resolution Representation Learning…
321138maintenance
arpitbansal297/Cold-Diffusion-Models
Official PyTorch implementation of Cold Diffusion, a research paper showing that diffusion-style generative models can invert arbitrary ima…
321136maintenance
karfly/learnable-triangulation-pytorch
Official PyTorch implementation of the ICCV 2019 paper 'Learnable Triangulation of Human Pose', providing algebraic and volumetric learnabl…
231135maintenance
harvardnlp/pytorch-struct
Torch-Struct is a PyTorch library of tested, GPU-accelerated implementations of core structured prediction algorithms such as CRFs, HMMs, H…
231133maintenance
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…
321132maintenance
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…
321132maintenance
fff-rs/juice
Juice is a Rust machine learning framework ('The Hacker's Machine Learning Engine') built on the Coaster hardware abstraction layer, suppor…
321132maintenance
MRzzm/DINet
DINet is the official PyTorch implementation of an AAAI 2023 paper on realistic face visually dubbing, which deforms and inpaints mouth reg…
321128maintenance
maudzung/SFA3D
A PyTorch implementation of SFA3D, a fast and accurate anchor-free 3D object detection model for LiDAR point clouds, trained and evaluated …
321128maintenance
KinWaiCheuk/nnAudio
nnAudio is a PyTorch-based audio processing toolbox that computes spectrograms (STFT, Mel, CQT, MFCC) using 1D convolutional neural network…
601127maintenance
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…
321127maintenance
yu-takagi/StableDiffusionReconstruction
Research codebase reproducing Takagi and Nishimoto's CVPR 2023 method for reconstructing images a person viewed from fMRI brain activity us…
301127maintenance
bearpaw/pytorch-pose
A PyTorch toolkit implementing a general pipeline for 2D single-human pose estimation, with training, inference, and evaluation interfaces …
321121maintenance
irolaina/FCRN-DepthPrediction
Reference implementation and pretrained models for FCRN (Deeper Depth Prediction with Fully Convolutional Residual Networks), predicting de…
321118maintenance
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 …
321117maintenance
snap-research/EfficientFormer
A PyTorch implementation of EfficientFormer and EfficientFormerV2, efficient vision transformer model families designed to run at MobileNet…
321116maintenance
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,…
231115maintenance
Res2Net/Res2Net-PretrainedModels
Official PyTorch implementation of Res2Net, a multi-scale CNN backbone architecture published in TPAMI, with ImageNet-pretrained model weig…
321114maintenance
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…
311111maintenance
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-…
271111maintenance
rosinality/style-based-gan-pytorch
A PyTorch implementation of the StyleGAN paper (A Style-Based Generator Architecture for Generative Adversarial Networks), including traini…
321110maintenance
andyzeng/visual-pushing-grasping
PyTorch reference implementation of Visual Pushing and Grasping (VPG), which trains robotic agents via self-supervised deep reinforcement l…
321109maintenance
fudan-zvg/SETR
SETR (SEgmentation TRansformers) is the official PyTorch implementation of the CVPR 2021 / IJCV 2024 paper 'Rethinking Semantic Segmentatio…
321108maintenance
novak-99/MLPP
ML++ is a C++ machine learning library providing regression models, dynamically sized neural networks, statistics, and optimization algorit…
231108maintenance
aim-uofa/AdelaiDepth
AdelaiDepth is an open-source toolbox for monocular depth prediction and 3D scene reconstruction from single images, containing research pr…
321106maintenance
baidu-research/DeepBench
DeepBench is a benchmarking suite from Baidu Research that measures the performance of fundamental deep learning operations (matrix multipl…
321106maintenance
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…
321104maintenance
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…
321104maintenance
keroro824/HashingDeepLearning
SLIDE is a C++ research codebase implementing locality-sensitive hashing based training of deep neural networks, from the paper 'In Defense…
321103maintenance
pfnet-research/sngan_projection
Official Chainer implementation of GANs using spectral normalization and projection discriminators for conditional image generation on Imag…
321103maintenance
PaddlePaddle/Paddle.js
Paddle.js is a browser-based deep learning inference engine for Baidu PaddlePaddle models, running via WebGL, WebGPU, or WebAssembly backen…
231103maintenance
fyu/drn
A PyTorch library implementing Dilated Residual Networks (DRN), which combine dilated convolutions with residual networks for image classif…
321102maintenance
hhaAndroid/mmdetection-mini
A minimal, heavily annotated reimplementation of the mmdetection object detection framework, built from scratch for learning purposes. It m…
321100maintenance
hongfz16/AvatarCLIP
Official PyTorch implementation of AvatarCLIP, a SIGGRAPH 2022 research framework that generates and animates 3D human avatars from natural…
321100maintenance
kazuto1011/deeplab-pytorch
An unofficial PyTorch re-implementation of DeepLab v2 with a ResNet-101 backbone for semantic segmentation, supporting COCO-Stuff and PASCA…
231100maintenance
jacobgil/vit-explain
A PyTorch library implementing explainability methods for Vision Transformers, including Attention Rollout and Gradient Attention Rollout. …
321098maintenance
NVlabs/NVAE
Official PyTorch implementation of NVAE, a deep hierarchical variational autoencoder for likelihood-based image generation from a NeurIPS 2…
321092maintenance
Tencent/TencentPretrain
TencentPretrain is a PyTorch-based toolkit for pre-training and fine-tuning transformer models across modalities such as text, vision, and …
321091maintenance
VITA-Group/EnlightenGAN
Official PyTorch implementation of EnlightenGAN, an IEEE TIP paper on deep low-light image enhancement without paired supervision using gen…
321091maintenance
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…
321089maintenance
yu4u/noise2noise
An unofficial, partial Keras implementation of the Noise2Noise paper for learning image restoration without clean training data. It trains …
231088maintenance
ethanhe42/channel-pruning
Reference implementation of the ICCV 2017 channel pruning method for accelerating very deep convolutional neural networks, using LASSO regr…
231088maintenance
sfzhang15/ATSS
Official PyTorch implementation of ATSS (Adaptive Training Sample Selection), a CVPR 2020 Oral paper on object detection. It automatically …
321086maintenance
GOATmessi8/ASFF
A PyTorch implementation of YOLOv3 with the Adaptively Spatial Feature Fusion (ASFF) module and optional MobileNetV2 backbone for single-sh…
321085maintenance
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. …
101085maintenance
locuslab/convmixer
An official PyTorch implementation of ConvMixer, a patch-based convolutional image classification architecture from the ICLR 2022 paper 'Pa…
231084maintenance
seongjunyun/Graph_Transformer_Networks
Authors' PyTorch implementation of Graph Transformer Networks (GTN) from the NeurIPS 2019 paper, plus FastGTN with non-local operations. It…
321083maintenance
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…
321083maintenance
rykov8/ssd_keras
A Keras port of the SSD (Single Shot MultiBox Detector) object detection model, including pretrained weights ported from the original Caffe…
101083maintenance
weiyithu/SurroundOcc
SurroundOcc is the official PyTorch implementation of an ICCV 2023 paper predicting dense 3D volumetric occupancy from multi-camera images …
431081maintenance
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…
321081maintenance
rixwew/pytorch-fm
A PyTorch library implementing a wide collection of factorization machine models (FM, FFM, DeepFM, xDeepFM, AutoInt, NFM, DCN, and more) fo…
321080maintenance
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…
321078maintenance
pytorch/torchdynamo
TorchDynamo is a Python-level JIT compiler that speeds up unmodified PyTorch programs by capturing Python bytecode into FX graphs. The proj…
101078maintenance
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…
321076maintenance
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…
321075maintenance
sunset1995/DirectVoxGO
DirectVoxGO (DVGO) is a PyTorch implementation of Direct Voxel Grid Optimization for fast neural radiance field (NeRF) reconstruction, repl…
321075maintenance
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…
321074maintenance
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…
321072maintenance
gusye1234/LightGCN-PyTorch
A PyTorch implementation of LightGCN, a simplified graph convolution network for collaborative filtering recommendation from the SIGIR 2020…
101072maintenance
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…
321071maintenance
MatthieuCourbariaux/BinaryNet
BinaryNet is a research codebase for training deep neural networks whose weights and activations are constrained to +1 or -1, reproducing t…
321068maintenance
openai/blocksparse
A Python package of TensorFlow ops and custom CUDA GPU kernels for block-sparse matrix multiplication and convolution, plus related ops lik…
101068maintenance
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 …
321067maintenance
omerbt/MultiDiffusion
Official PyTorch implementation of MultiDiffusion (ICML 2023), a training-free framework that fuses multiple diffusion paths over a pre-tra…
311066maintenance
megvii-research/PETR
Official PyTorch implementation of PETR and PETRv2, transformer-based frameworks for multi-view 3D object detection and BEV segmentation fr…
321065maintenance
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…
231065maintenance
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…
321063maintenance
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…
231063maintenance
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…
291062maintenance
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…
321061maintenance
wubinzzu/NeuRec
NeuRec is an open-source Python library for recommender systems built on TensorFlow, offering 33 state-of-the-art neural recommendation mod…
231061maintenance
jsyoon0823/TimeGAN
Reference implementation of TimeGAN, a generative adversarial network framework for generating synthetic time-series data, published at Neu…
621060maintenance
OFA-Sys/ONE-PEACE
ONE-PEACE is a general multimodal representation model that jointly encodes vision, audio, and language modalities without initializing fro…
291060maintenance
philipperemy/keract
Keract is a Python library that extracts layer activations (outputs) and gradients from TensorFlow/Keras models as NumPy arrays, with helpe…
301059maintenance
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…
231059maintenance
chenyuntc/PyTorchText
A PyTorch implementation of multiple text classification models (TextCNN, TextRNN/LSTM, RCNN, FastText, inception CNN) that won 1st place i…
321057maintenance
HRNet/HRNet-Image-Classification
Official PyTorch implementation and training code for HRNet (High-Resolution Network) image classification models on ImageNet. It provides …
231056maintenance
RexYing/gnn-model-explainer
Official implementation of GNNExplainer, a NeurIPS 2019 method for explaining predictions of graph neural networks by learning soft masks o…
321055maintenance
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 …
321054maintenance
raminmh/CfC
Reference implementations of Closed-form Continuous-time (CfC) neural networks, a fast closed-form approximation of liquid time-constant ne…
231053maintenance

← prev page 20 / 27 next →