A PyTorch implementation of EfficientNet

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.

StyleGAN2-ADA - Official PyTorch implementation

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX

CNN Image Retrieval in PyTorch: Training and evaluating CNNs for Image Retrieval in PyTorch

Designing a Practical Degradation Model for Deep Blind Image Super-Resolution (ICCV, 2021) (PyTorch) - We released the training code!

Implementation of NIMA: Neural Image Assessment in Keras

Dense matching library based on PyTorch

Compiler for LightGBM gradient-boosted trees, based on LLVM. Speeds up prediction by ≥10x.

Official PyTorch implementation of "ArtFlow: Unbiased Image Style Transfer via Reversible Neural Flows"