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

domain: deep-learning

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

ProductHealth v2StarsMaturity
PlayVoice/whisper-vits-svc
A PyTorch-based singing voice conversion and voice cloning engine built on VITS with Whisper, BigVGAN, and diffusion components. It lets us…
232864maintenance
kpzhang93/MTCNN_face_detection_alignment
Reference MATLAB/Caffe implementation of MTCNN, a multi-task cascaded convolutional neural network for joint face detection and facial land…
322862maintenance
gwding/draw_convnet
A Python script that generates illustrations of Convolutional Neural Network (ConvNet) architectures using matplotlib. It produces publicat…
322860maintenance
cybertronai/gradient-checkpointing
A Python library that reduces GPU memory usage when training very deep neural networks via gradient checkpointing, trading computation for …
322843maintenance
IDEA-Research/DINO
Official PyTorch implementation of DINO, a state-of-the-art end-to-end object detection model based on DETR with improved denoising anchor …
322834maintenance
rosinality/stylegan2-pytorch
A PyTorch implementation of StyleGAN 2, the image-quality-improved version of the StyleGAN generative adversarial network, including traini…
322833maintenance
kzl/decision-transformer
Official codebase for the Decision Transformer paper, which frames reinforcement learning as sequence modeling with transformers. It contai…
322830maintenance
jina-ai/dalle-flow
DALL·E Flow is a client-server workflow for generating high-definition images from text prompts using DALL·E-Mega, GLID-3 XL, and Stable Di…
322829maintenance
huggingface/knockknock
A small Python library that sends notifications when machine learning training completes or crashes, requiring only a decorator on your mai…
322828maintenance
ai-forever/Kandinsky-2
Kandinsky 2 is a multilingual text-to-image latent diffusion model library from Sber AI, supporting text2img, img2img, inpainting, outpaint…
322814maintenance
SCUTlihaoyu/open-chat-video-editor
An open-source Python tool that automatically generates short videos from a short text prompt or a web URL, producing narration, background…
292813maintenance
philipperemy/keras-attention
A Keras layer implementing attention mechanisms with Luong (multiplicative) and Bahdanau (additive) score functions for TensorFlow 2.x. It …
552809maintenance
roytseng-tw/Detectron.pytorch
A PyTorch reimplementation of Facebook's Detectron object detection framework, supporting Mask R-CNN, keypoint/pose estimation, and instanc…
102808maintenance
kwea123/nerf_pl
An unofficial PyTorch implementation of NeRF (Neural Radiance Fields) and NeRF in the Wild built on pytorch-lightning, supporting multi-GPU…
232805maintenance
Alpha-VLLM/LLaMA2-Accessory
LLaMA2-Accessory is an open-source Python toolkit for pretraining, finetuning, and deploying large language models and multimodal LLMs, inc…
292800maintenance
PhoebusSi/Alpaca-CoT
Alpaca-CoT is an instruction-tuning platform that unifies interfaces for instruction data collection, parameter-efficient fine-tuning metho…
302791maintenance
microsoft/NUWA
Microsoft's official research repository for the NUWA family of multimodal generative models, a unified 3D transformer pipeline for visual …
102791maintenance
ethereon/caffe-tensorflow
A Python tool that converts Caffe-trained neural network models into TensorFlow format, producing a NumPy parameter file and a Python class…
322786maintenance
henrywoo/pyllama
A Python library that is a modified version of Meta's LLaMA implementation, making it easy to download model weights and run LLaMA models o…
212778maintenance
apple/ml-ane-transformers
A reference PyTorch implementation of the Transformer architecture optimized for Apple's Neural Engine (ANE), including optimized Hugging F…
232736maintenance
cbfinn/maml
Reference implementation of Model-Agnostic Meta-Learning (MAML) from Finn et al., ICML 2017, for fast adaptation of deep networks. It inclu…
322727maintenance
kyegomez/OpenMythos
OpenMythos is an open-source, theoretical PyTorch implementation of a Recurrent-Depth Transformer (RDT) architecture inspired by speculatio…
5114805experimental
carpedm20/ENAS-pytorch
A PyTorch implementation of Efficient Neural Architecture Search (ENAS), which reduces NAS GPU-hours by 1000x via parameter sharing between…
322719maintenance
torch-points3d/torch-points3d
A PyTorch-based framework for deep learning on 3D point clouds, supporting models like PointNet, KPConv, and MinkowskiEngine sparse convolu…
672711maintenance
zllrunning/video-object-removal
A PyTorch application that removes objects from videos by drawing a bounding box around them. It combines SiamMask for object tracking and …
322711maintenance
junyanz/interactive-deep-colorization
A deep learning application for interactively colorizing black and white images with user-guided hints, based on the SIGGRAPH 2017 paper 'R…
322692maintenance
mahyarnajibi/SNIPER
SNIPER is an efficient multi-scale training algorithm for object detection and instance segmentation that processes only context regions (c…
322690maintenance
pengzhiliang/MAE-pytorch
An unofficial PyTorch implementation of the MAE (Masked Autoencoders Are Scalable Vision Learners) paper, providing pretraining and finetun…
322689maintenance
microsoft/pai
OpenPAI is an open-source AI platform from Microsoft that provides resource scheduling and cluster management for machine learning workload…
662688maintenance
JIA-Lab-research/LongLoRA
LongLoRA is an efficient fine-tuning approach and codebase that extends the context length of pre-trained LLMs (Llama2 7B/13B/70B) using sh…
272686maintenance
graphdeeplearning/benchmarking-gnns
An open-source benchmarking framework for graph neural networks (GNNs) published as a JMLR 2023 paper. It provides reproducible pipelines, …
322671maintenance
yerfor/GeneFace
GeneFace is the official PyTorch implementation of an ICLR 2023 paper on generalized, high-fidelity audio-driven 3D talking face synthesis …
212657maintenance
KupynOrest/DeblurGAN
A PyTorch implementation of the DeblurGAN paper for blind motion deblurring using conditional adversarial networks. It uses a Conditional W…
322638maintenance
HypoX64/DeepMosaics
DeepMosaics is a Python application that automatically removes or adds mosaics in images and videos using semantic segmentation and image-t…
232633maintenance
sanghyun-son/EDSR-PyTorch
Official PyTorch implementation of EDSR (Enhanced Deep Residual Networks for Single Image Super-Resolution, CVPRW 2017). It provides traini…
322627maintenance
knazeri/edge-connect
EdgeConnect is a PyTorch implementation of a two-stage generative adversarial model for image inpainting, published at ICCV 2019. It first …
322620maintenance
PeterH0323/Smart_Construction
A YOLOv5-based object detection application for detecting people, heads, and safety helmets on construction sites, including pretrained wei…
232619maintenance
yangxy/GPEN
GPEN (GAN Prior Embedded Network) is a deep learning library for blind face restoration of low-quality, in-the-wild photos. It also support…
642612maintenance
baaivision/Painter
Painter and SegGPT are vision foundation models from BAAI for in-context visual learning, where a single generalist model performs diverse …
322593maintenance
ShawnBIT/UNet-family
A curated collection of UNet-family semantic segmentation models with PyTorch implementations and links to original papers and third-party …
322592maintenance
lucidrains/big-sleep
Big Sleep is a Python command line tool that generates images from natural language text prompts by combining OpenAI's CLIP with a BigGAN g…
232573maintenance
MaybeShewill-CV/lanenet-lane-detection
An unofficial TensorFlow implementation of the LaneNet deep neural network for real-time lane detection, based on the IEEE IV paper 'Toward…
322562maintenance
s3prl/s3prl
S3PRL is a PyTorch toolkit for self-supervised speech pre-training and representation learning, bundling many upstream models like wav2vec …
552561maintenance
tg-bomze/Face-Depixelizer
A Jupyter Notebook-based tool that turns low-resolution pixelated face images into realistic high-resolution faces using the PULSE method, …
322560maintenance
jcjohnson/torch-rnn
torch-rnn provides high-performance, reusable RNN and LSTM modules for the Torch7 deep learning framework, used for character-level languag…
322559maintenance
onnx/tensorflow-onnx
tf2onnx is a tool that converts TensorFlow, Keras, TensorFlow.js, and TFLite models into the ONNX format via a command-line interface or Py…
772553maintenance
LiyuanLucasLiu/RAdam
RAdam is a Python implementation of Rectified Adam, a variant of the Adam optimizer that analytically reduces the large variance of adaptiv…
322549maintenance
taesungp/contrastive-unpaired-translation
A PyTorch implementation of Contrastive Unpaired Translation (CUT), an ECCV 2020 method for unpaired image-to-image translation using patch…
322543maintenance
automl/Auto-PyTorch
Auto-PyTorch is an AutoML framework that jointly performs neural architecture search and hyperparameter optimization for PyTorch models, us…
232541maintenance
dmlc/gluon-nlp
GluonNLP is a Python toolkit built on Apache MXNet that provides text data loading, processing tools, pretrained model zoos, and training u…
102540maintenance
znxlwm/UGATIT-pytorch
Official PyTorch implementation of U-GAT-IT, an unsupervised image-to-image translation model using attention modules and adaptive layer-in…
322536maintenance
google-deepmind/dnc
A TensorFlow implementation of the Differentiable Neural Computer (DNC), a recurrent neural network architecture with an external memory pu…
322531maintenance
young-geng/EasyLM
EasyLM is a JAX/Flax-based framework for pre-training, finetuning, evaluating, and serving large language models like LLaMA. It scales trai…
322514maintenance
dmlc/decord
Decord is a C++ library with Python bindings that provides efficient video and audio decoding for deep learning pipelines. It wraps hardwar…
232514maintenance
zzh8829/yolov3-tf2
A clean implementation of YOLOv3 and YOLOv3-tiny object detection in TensorFlow 2.0, with pre-trained Darknet weight conversion, inference,…
322513maintenance
cure-lab/LTSF-Linear
Official PyTorch implementation of LTSF-Linear (Linear, DLinear, NLinear) from the AAAI-23 paper 'Are Transformers Effective for Time Serie…
322508maintenance
meijieru/crnn.pytorch
A PyTorch implementation of the Convolutional Recurrent Neural Network (CRNN) for scene text recognition, based on the 2016 paper by Shi et…
322492maintenance
galeone/tfgo
tfgo is a Go library that wraps TensorFlow's Go bindings with a friendlier, method-chaining API for building and executing computation grap…
322491maintenance
dragen1860/MAML-Pytorch
A PyTorch implementation of Model-Agnostic Meta-Learning (MAML) for few-shot supervised learning, supporting MiniImagenet and Omniglot benc…
322483maintenance
microsoft/Graphormer
Graphormer is a deep learning library from Microsoft providing a graph transformer backbone for molecular modeling tasks. It ships pre-trai…
622474maintenance
CASIA-IVA-Lab/DANet
DANet is the official PyTorch implementation of 'Dual Attention Network for Scene Segmentation' (CVPR 2019), which uses position and channe…
322463maintenance
XifengGuo/CapsNet-Keras
A Keras implementation of CapsNet, the capsule network architecture from the NIPS 2017 paper 'Dynamic Routing Between Capsules', achieving …
232456maintenance
Zhongdao/Towards-Realtime-MOT
A PyTorch codebase for the Joint Detection and Embedding (JDE) model, a fast multiple-object tracker that learns object detection and appea…
322444maintenance
HobbitLong/RepDistiller
RepDistiller is a PyTorch research codebase implementing Contrastive Representation Distillation (CRD) from ICLR 2020. It also benchmarks 1…
322437maintenance
CyberZHG/keras-bert
A Keras implementation of BERT that can load official Google pre-trained checkpoints for feature extraction and prediction. It includes a t…
102419maintenance
igul222/improved_wgan_training
Reference implementation of the 'Improved Training of Wasserstein GANs' (WGAN-GP) paper, providing TensorFlow code to reproduce its experim…
322411maintenance
OpenBMB/CPM-Bee
CPM-Bee is a fully open-source, commercially usable 10-billion-parameter bilingual (Chinese-English) foundation large language model traine…
712406maintenance
deepcam-cn/yolov5-face
YOLOv5-Face is a real-time, high-accuracy face detector built on the YOLOv5 object detection framework in PyTorch, with TensorRT deployment…
322406maintenance
mravanelli/pytorch-kaldi
PyTorch-Kaldi is a toolkit for developing state-of-the-art DNN/HMM hybrid speech recognition systems, combining PyTorch-managed neural netw…
322404maintenance
danielegrattarola/spektral
Spektral is a Python library for graph deep learning built on the Keras API and TensorFlow 2. It provides popular graph neural network laye…
232396maintenance
iPERDance/iPERCore
Impersonator++ (iPERCore) is a PyTorch implementation of Liquid Warping GAN with Attention, a unified framework for human image synthesis. …
322393maintenance
asyml/texar
Texar is a modularized Python toolkit for machine learning, especially natural language processing and text generation, built on TensorFlow…
652389maintenance
BrikerMan/Kashgari
Kashgari is a Keras/TensorFlow 2-based NLP transfer learning framework for building text labeling (NER, PoS) and text classification models…
232381maintenance
r9y9/wavenet_vocoder
A PyTorch implementation of the WaveNet vocoder that generates high-quality raw speech waveforms conditioned on acoustic features like mel-…
232376maintenance
nbei/Deep-Flow-Guided-Video-Inpainting
A PyTorch implementation of the CVPR 2019 paper 'Deep Flow-Guided Video Inpainting', which fills missing regions in videos by completing op…
322375maintenance
princeton-vl/CornerNet
Official research code for CornerNet, an object detection model that detects objects as paired keypoints, reproducing results from the ECCV…
322369maintenance
google-research/electra
ELECTRA is a research library from Google for self-supervised pre-training of transformer text encoders using a discriminator-based objecti…
102367maintenance
IST-DASLab/gptq
Reference implementation of GPTQ, a one-shot post-training weight quantization method for large generative transformer models based on appr…
322360maintenance
michuanhaohao/reid-strong-baseline
A PyTorch implementation of the 'Bag of Tricks and A Strong Baseline for Deep Person Re-identification' paper (CVPRW 2019), providing end-t…
322355maintenance
moskomule/senet.pytorch
A PyTorch implementation of Squeeze-and-Excitation Networks (SENet), the ILSVRC 2017 classification winner, including SE-ResNet and SE-Ince…
232344maintenance
smallcorgi/Faster-RCNN_TF
A TensorFlow implementation of Faster R-CNN, a convolutional neural network for object detection with a region proposal network. It include…
322342maintenance
NVIDIA/waveglow
WaveGlow is a PyTorch implementation of a flow-based generative network that synthesizes high-quality speech audio from mel-spectrograms, c…
322339maintenance
salesforce/decaNLP
decaNLP is a PyTorch research framework and benchmark for the Natural Language Decathlon, a multitask NLP challenge spanning ten tasks from…
102337maintenance
Rayhane-mamah/Tacotron-2
A TensorFlow implementation of DeepMind's Tacotron-2 neural text-to-speech architecture, including both the Tacotron spectrogram predictor …
322322maintenance
hzwer/ICCV2019-LearningToPaint
A PyTorch research implementation of the ICCV 2019 paper 'Learning to Paint With Model-based Deep Reinforcement Learning'. It trains agents…
412305maintenance
AIStream-Peelout/flow-forecast
Flow Forecast is a PyTorch-based deep learning framework for time series forecasting, classification, and anomaly detection. It offers stat…
662292maintenance
JonathonLuiten/Dynamic3DGaussians
Official PyTorch implementation of 'Dynamic 3D Gaussians: Tracking by Persistent Dynamic View Synthesis' (3DV 2024), which models dynamic 3…
282292maintenance
titu1994/Neural-Style-Transfer
A Keras 2.0+ implementation of Neural Style Transfer from the paper 'A Neural Algorithm of Artistic Style', with improvements like INetwork…
232287maintenance
jsbroks/coco-annotator
COCO Annotator is a web-based image annotation tool for labeling images with segments, bounding boxes, keypoints, and object tracking to cr…
252279maintenance
astooke/rlpyt
A PyTorch library providing modular, high-throughput implementations of deep reinforcement learning algorithms, covering policy gradient, d…
322278maintenance
zju3dv/NeuralRecon
NeuralRecon is a deep learning framework for real-time 3D scene reconstruction from monocular video with known camera poses. It reconstruct…
322274maintenance
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…
292268maintenance
andreas128/RePaint
Official PyTorch implementation of RePaint, a CVPR 2022 paper on image inpainting using denoising diffusion probabilistic models. It provid…
322267maintenance
ndrplz/ConvLSTM_pytorch
A PyTorch implementation of the Convolutional LSTM (ConvLSTM) neural network module, provided as a reusable nn.Module supporting arbitrary …
322267maintenance
mrharicot/monodepth
A TensorFlow implementation of unsupervised monocular depth estimation from single images using convolutional neural networks, based on the…
322265maintenance
namisan/mt-dnn
A PyTorch library implementing Multi-Task Deep Neural Networks (MT-DNN) for natural language understanding, based on Microsoft research pap…
322257maintenance
ShoufaChen/DiffusionDet
PyTorch implementation of DiffusionDet, the first diffusion-model-based object detection framework (ICCV 2023 Best Paper Finalist). It prov…
222257maintenance
Tencent/MedicalNet
MedicalNet provides a series of 3D-ResNet pre-trained models trained on 23 diverse medical imaging datasets, with PyTorch transfer-learning…
572255maintenance
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…
322254maintenance
donnyyou/torchcv
TorchCV is a PyTorch-based framework providing reimplementations of deep learning models for major computer vision tasks. It covers image c…
322251maintenance

← prev page 16 / 28 next →