Official Implementation for "Encoding in Style: a StyleGAN Encoder for Image-to-Image Translation" (CVPR 2021) presenting the pixel2style2pixel (pSp) framework

POT : Python Optimal Transport

A PyTorch reimplementation of the paper Free-Form Image Inpainting with Gated Convolution (DeepFill v2) (https://arxiv.org/abs/1806.03589)

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also …

Toolbox of models, callbacks, and datasets for AI/ML researchers.

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

Hydra is a framework for elegantly configuring complex applications

PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡

Compute FID scores with PyTorch.

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

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

Implementation of "Improving the Improved Training of Wasserstein GANs: A Consistency Term and Its Dual Effect" in pytorch