Search Packages

clean-fid

PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]

piq

Measures and metrics for image2image tasks. PyTorch.

gfpgan

GFPGAN aims at developing Practical Algorithms for Real-world Face Restoration.

gluoncv

Gluon CV Toolkit

sdv

Synthetic data generation for tabular data

pytorch-fid

Compute FID scores with PyTorch.

lightning-bolts

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

so-vits-svc-fork

so-vits-svc fork with realtime support, improved interface and more features.

ydata-synthetic

Synthetic data generators for tabular and time-series data

nnm

Neural Network Models

torchstudio

Deep Learning Experiment

socaity-face2face

Swap faces in images and videos. Create face embeddings. Enhance face image quality. Deploy as a web api.

tensorlayer

Deep Learning and Reinforcement Learning Library for Scientists and Engineers

labml-nn

🧑‍🏫 Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit), optimizers (adam, radam, adabelief), gans(dcgan, cyclegan, stylegan2), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, diffusion, etc. 🧠

pytorch-pretrained-biggan

🦋A PyTorch implementation of BigGAN with pretrained weights and conversion scripts.

focal-frequency-loss

[ICCV 2021] Focal Frequency Loss for Image Reconstruction and Synthesis

neuralnetlib

A flexible deep learning framework built from scratch using only NumPy

pytorch-lightning-bolts

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

phaseaug

ICASSP 2023 Accepted

sdgx

SDG is a specialized framework designed to generate high-quality structured tabular data.

supercongan

Training a GAN using superconductivity data

hypergan

A customizable generative adversarial network with reproducible configurations. Build your own content generator.

dgmr

Implementation of DeepMind's Deep Generative Model of Radar (DGMR) https://arxiv.org/abs/2104.00954

stylegan2-torch

StyleGAN2 Pytorch - Typed, Commented, Installable :)