The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

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

Official pytorch implementation of the paper: "SinGAN: Learning a Generative Model from a Single Natural Image"

Unofficial implementation of Palette: Image-to-Image Diffusion Models by Pytorch

PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet

A flexible package for multimodal-deep-learning to combine tabular data with text and images using Wide and Deep models in Pytorch

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

Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI

Implementation of CoCa, Contrastive Captioners are Image-Text Foundation Models, in Pytorch

Official PyTorch implementation of ODISE: Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Models [CVPR 2023 Highlight]

[CVPR 2022] Official PyTorch Implementation for DiffusionCLIP: Text-guided Image Manipulation Using Diffusion Models

PyTorch implementation of a 1.3B text-to-image generation model trained on 14 million image-text pairs

:fire: Use pre-trained models in PyTorch to extract vector embeddings for any image

The pytorch implementation of our CVPR 2023 paper "Conditional Image-to-Video Generation with Latent Flow Diffusion Models"

A modified reimplemented in pytorch of inpainting model in Free-Form Image Inpainting with Gated Convolution [http://jiahuiyu.com/deepfill2/]

Official Pytorch Implementation of the paper: Wavelet Diffusion Models are fast and scalable Image Generators (CVPR'23)

Pytorch code to construct a 3D point cloud model from single RGB image.

Official PyTorch implementation for the paper "AnimateZero: Video Diffusion Models are Zero-Shot Image Animators"

A Flexible and Unified Image Restoration Framework (PyTorch), including state-of-the-art image restoration model. Such as NAFNet, Restormer, MPRNet, MIMO-UNet, SCUNet, SwinIR, HINet, etc. ⭐⭐⭐⭐⭐⭐

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".

PyTorch implementation of the ICCV paper "3D-aware Image Generation using 2D Diffusion Models"

Official pytorch implementation for "Low-light Image Enhancement with Wavelet-based Diffusion Models"

This is the pytorch implement of the paper "RSMamba: Remote Sensing Image Classification with State Space Model"

Implementation of Siamese Networks for image one-shot learning by PyTorch, train and test model on dataset Omniglot

TensorFlow port of PyTorch Image Models (timm) - image models with pretrained weights.

Official pytorch implementation for "LightenDiffusion: Unsupervised Low-Light Image Enhancement with Latent-Retinex Diffusion Models"

This is a warehouse for MobileNetV4-Pytorch-model, can be used to train your image-datasets for vision tasks.

This YOLOv5🚀😊 GUI road sign system uses MySQL💽, PyQt5🎨, PyTorch, CSS🌈. It has modules for login🔑, YOLOv5 setup📋, sign recognition🔍, database💾, and image processing🖼️. It supports diverse inputs, model switching, and enhancements like mosaic and mixup📈.

Pytorch implementation of "SinGAN: Learning a Generative Model from a Single Natural Image"

PyTorch implementation of the model presented in "Satellite Image Time Series Classification with Pixel-Set Encoders and Temporal Self-Attention"