shakediel

fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

pytorch-toolbelt

PyTorch extensions for fast R&D prototyping and Kaggle farming

PaddleVideo

Awesome video understanding toolkits based on PaddlePaddle. It supports video data annotation tools, lightweight RGB and skeleton based action recognition model, practical applications for video tagging and sport action detection.

evotorch

Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.

stable-diffusion

A latent text-to-image diffusion model

awesome-tips

techniques

Techniques for deep learning with satellite & aerial imagery

imagen-pytorch

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch

Neural-Scene-Flow-Fields

PyTorch implementation of paper "Neural Scene Flow Fields for Space-Time View Synthesis of Dynamic Scenes"

nerf_pl

NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

avalanche

Avalanche: an End-to-End Library for Continual Learning based on PyTorch.

NeRFusion

binance-trade-bot

Automated cryptocurrency trading bot

video-diffusion-pytorch

Implementation of Video Diffusion Models, Jonathan Ho's new paper extending DDPMs to Video Generation - in Pytorch

DALLE2-pytorch

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

OWOD

(CVPR 2021 Oral) Open World Object Detection

yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

FinRL

FinRL: Financial Reinforcement Learning. 🔥

stylegan3-editing

Official Implementation of "Third Time's the Charm? Image and Video Editing with StyleGAN3" (AIM ECCVW 2022) https://arxiv.org/abs/2201.13433

sahi

Framework agnostic sliced/tiled inference + interactive ui + error analysis plots

evojax

VQGAN-CLIP

Just playing with getting VQGAN+CLIP running locally, rather than having to use colab.

cflow-ad

Official PyTorch code for WACV 2022 paper "CFLOW-AD: Real-Time Unsupervised Anomaly Detection with Localization via Conditional Normalizing Flows"

cs-flow

This is the official repository to the WACV 2022 paper "Fully Convolutional Cross-Scale-Flows for Image-based Defect Detection" by Marco Rudolph, Tom Wehrbein, Bodo Rosenhahn and Bastian Wandt.

SRFlow

Official SRFlow training code: Super-Resolution using Normalizing Flow in PyTorch

RePaint

Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022

STIT