pytorch

PeterL1n/RobustVideoMatting

PeterL1n/RobustVideoMatting

November 4, 2021

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!
PyTorchLightning/lightning-flash

PyTorchLightning/lightning-flash

June 11, 2021

Collection of tasks for fast prototyping, baselining, finetuning and solving problems with deep learning.
facebookresearch/dino

facebookresearch/dino

April 30, 2021

PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
jacobgil/pytorch-grad-cam

jacobgil/pytorch-grad-cam

April 28, 2021

Many Class Activation Map methods implemented in Pytorch for CNNs and Vision Transformers. Including Grad-CAM, Grad-CAM++, Score-CAM, Ablation-CAM and XGrad-CAM
zihangJiang/TokenLabeling

zihangJiang/TokenLabeling

April 28, 2021

Pytorch implementation of "Training a 85.4% Top-1 Accuracy Vision Transformer with 56M Parameters on ImageNet"
facebookresearch/pytorchvideo

facebookresearch/pytorchvideo

April 14, 2021

A deep learning library for video understanding research.
bes-dev/MobileStyleGAN.pytorch

bes-dev/MobileStyleGAN.pytorch

April 9, 2021

An official implementation of MobileStyleGAN in PyTorch
BenAAndrew/Voice-Cloning-App

BenAAndrew/Voice-Cloning-App

April 2, 2021

A Python/Pytorch app for easily synthesising human voices
aladdinpersson/Machine-Learning-Collection

aladdinpersson/Machine-Learning-Collection

March 26, 2021

A resource for learning about ML, DL, PyTorch and TensorFlow. Feedback always appreciated :)
berniwal/swin-transformer-pytorch

berniwal/swin-transformer-pytorch

March 26, 2021

Implementation of the Swin Transformer in PyTorch.
facebookresearch/barlowtwins

facebookresearch/barlowtwins

March 19, 2021

PyTorch implementation of Barlow Twins.
jadore801120/attention-is-all-you-need-pytorch

jadore801120/attention-is-all-you-need-pytorch

March 12, 2021

A PyTorch implementation of the Transformer model in "Attention is All You Need".
rosinality/stylegan2-pytorch

rosinality/stylegan2-pytorch

March 12, 2021

Implementation of Analyzing and Improving the Image Quality of StyleGAN (StyleGAN 2) in PyTorch
speechbrain/speechbrain

speechbrain/speechbrain

March 12, 2021

A PyTorch-based Speech Toolkit
pytorch/opacus

pytorch/opacus

March 11, 2021

Training PyTorch models with differential privacy
Yuheng-Li/MixNMatch

Yuheng-Li/MixNMatch

March 11, 2021

Pytorch implementation of MixNMatch
AI4Finance-LLC/ElegantRL

AI4Finance-LLC/ElegantRL

March 9, 2021

Lightweight, efficient and stable implementations of deep reinforcement learning algorithms using PyTorch.
POSTECH-CVLab/PyTorch-StudioGAN

POSTECH-CVLab/PyTorch-StudioGAN

February 26, 2021

StudioGAN is a Pytorch library providing implementations of representative Generative Adversarial Networks (GANs) for conditional/unconditional image generation.
lucidrains/DALLE-pytorch

lucidrains/DALLE-pytorch

February 25, 2021

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch
openai/DALL-E

openai/DALL-E

February 25, 2021

PyTorch package for the discrete VAE used for DALLE.
yongxinw/GSDT

yongxinw/GSDT

February 25, 2021

Official PyTorch implementation of "Joint Object Detection and Multi-Object Tracking with Graph Neural Networks"
NVlabs/stylegan2-ada-pytorch

NVlabs/stylegan2-ada-pytorch

February 5, 2021

StyleGAN2-ADA - Official PyTorch implementation