owenzlz

dpdmd. [ICML 2026] The offical code of Diversity-Preserved Distribution Matching Distillation for Fast Visual Synthesis

amodal. Official code for "Amodal Completion via Progressive Mixed Context Diffusion" [CVPR 2024 Highlight]

FeatUp. Official code for "FeatUp: A Model-Agnostic Frameworkfor Features at Any Resolution" ICLR 2024

LayerDiffuse. Transparent Image Layer Diffusion using Latent Transparency

SoM. [arXiv 2023] Set-of-Mark Prompting for GPT-4V and LMMs

Depth-Anything. [CVPR 2024] Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data. Foundation Model for Monocular Depth Estimation

GLEE. [CVPR2024 Highlight]GLEE: General Object Foundation Model for Images and Videos at Scale

ViP-LLaVA. [CVPR2024] ViP-LLaVA: Making Large Multimodal Models Understand Arbitrary Visual Prompts

LVM. Python

Improving-Lens-Flare-Removal. Official implementation of ICCV 2023 Improving Lens Flare Removal with General-Purpose Pipeline and Multiple Light Sources Recovery

DINOv. [CVPR 2024] Official implementation of the paper "Visual In-context Learning"

prompt-to-prompt. Jupyter Notebook

PAL4VST. Perceptual Artifacts Localization for Image Synthesis Tasks (ICCV 23')

MiniGPT-4. Open-sourced codes for MiniGPT-4 and MiniGPT-v2 (https://minigpt-4.github.io, https://minigpt-v2.github.io/)

MasaCtrl. [ICCV 2023] Consistent Image Synthesis and Editing

Caption-Anything. Caption-Anything is a versatile tool combining image segmentation, visual captioning, and ChatGPT, generating tailored captions with diverse controls for user preferences. https://huggingface.co/spaces/TencentARC/Caption-Anything https://huggingface.co/spaces/VIPLab/Caption-Anything

EgoHOS. Fine-Grained Egocentric Hand-Object Segmentation, ECCV 2022

Grounded-Diffusion. Open-vocabulary Object Segmentation with Diffusion Models

dalle-playground. A playground to generate images from any text prompt using Stable Diffusion (past: using DALL-E Mini)

Awesome-Diffusion-Models. A collection of resources and papers on Diffusion Models

Paint-by-Example. Paint by Example: Exemplar-based Image Editing with Diffusion Models

InstaOrder. Instance-wise Occlusion and Depth Orders in Natural Scenes (CVPR 2022)

Dreambooth-Stable-Diffusion. Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion

gradio. Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

textual_inversion. Jupyter Notebook

DeepImageBlending. This is a Pytorch implementation of deep image blending

PAL4Inpaint. Perceptual Artifacts Localization for Inpainting, ECCV 2022 (Oral)

SuperCAF. Inpainting at Modern Camera Resolution by Guided PatchMatch with Auto-Curation, ECCV 2022

stargan-v2. StarGAN v2 - Official PyTorch Implementation (CVPR 2020)

stable-diffusion. A latent text-to-image diffusion model

STIT. Python

transfiner. Mask Transfiner for High-Quality Instance Segmentation, CVPR 2022

RefineMask. RefineMask: Towards High-Quality Instance Segmentation with Fine-Grained Features (CVPR 2021)

parti.

CM-GAN-Inpainting. CM-GAN for Image Inpainting

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

co-mod-gan-pytorch. pytorch implementation of the paper ``Large Scale Image Completion via Co-Modulated Generative Adversarial Networks"

ClickSEG. A code base for interactive segmentation

stylegan2-pytorch. Simplest working implementation of Stylegan2, state of the art generative adversarial network, in Pytorch. Enabling everyone to experience disentanglement

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

DeepGrabCut-PyTorch. Deep GrabCut in PyTorch

CLIP_prefix_caption. Simple image captioning model

stylegan3. Official PyTorch implementation of StyleGAN3

pose-with-style. [SIGGRAPH Asia 2021] Pose with Style: Detail-Preserving Pose-Guided Image Synthesis with Conditional StyleGAN

latent-diffusion. High-Resolution Image Synthesis with Latent Diffusion Models

baktsiu. An image viewer designed for comparing images and examining pixel differences

OpenAI-CLIP. Simple implementation of OpenAI CLIP model in PyTorch.

webpage-template. Simple project webpage template. Originally used in Colorful Image Colorization. ECCV, 2016.

handAR. Python

Image-Super-Resolution-via-Iterative-Refinement. Unofficial implementation of Image Super-Resolution via Iterative Refinement by Pytorch

C2-Action-Detection. Evaluation measures for the EPIC-KITCHENS-100 Action Detection challenge

epic-kitchens-slowfast. Python

CLIP. CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image

VIBE. Official implementation of CVPR2020 paper "VIBE: Video Inference for Human Body Pose and Shape Estimation"

homan. [3dv 2021] Joint fitting of hands and object from short RGB video clips

CascadePSP. [CVPR 2020] CascadePSP: Toward Class-Agnostic and Very High-Resolution Segmentation via Global and Local Refinement

frankmocap. A Strong and Easy-to-use Single View 3D Hand+Body Pose Estimator

LETR. (CVPR 2021 Oral) LETR: Line Segment Detection Using Transformers without Edges

pytorchvideo. A deep learning library for video understanding research.

DPT. Dense Prediction Transformers

lama. 🦙 LaMa Image Inpainting, Resolution-robust Large Mask Inpainting with Fourier Convolutions, WACV 2022

Real-ESRGAN. Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.

sketcher. Photo to sketch effects in R

stylegan2-ada-pytorch. StyleGAN2-ADA - Official PyTorch implementation

TransGAN. [NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang

pytorch-image-models. 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

co-mod-gan. [ICLR 2021, Spotlight] Large Scale Image Completion via Co-Modulated Generative Adversarial Networks

Swin-Transformer. This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

Awesome-Visual-Transformer. Collect some papers about transformer with vision. Awesome Transformer with Computer Vision (CV)

swin-transformer-pytorch. Implementation of the Swin Transformer in PyTorch.

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

latent-composition. Python

ContactPose. Large dataset of hand-object contact, hand- and object-pose, and 2.9 M RGB-D grasp images.

hand_dapg. Repository to accompany RSS 2018 paper on dexterous hand manipulation

dsntnn. PyTorch implementation of DSNT

stylegan2-pytorch. Implementation of Analyzing and Improving the Image Quality of StyleGAN (StyleGAN 2) in PyTorch

video-object-removal. Just draw a bounding box and you can remove the object you want to remove.

HRNet-Semantic-Segmentation. The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919

interaction-exploration. Code for "Learning Affordance Landscapes for Interaction Exploration in 3D Environments" (NeurIPS 20)

SparseR-CNN. [CVPR2021, PAMI2023] End-to-End Object Detection with Learnable Proposal

GANLatentDiscovery. The authors official implementation of Unsupervised Discovery of Interpretable Directions in the GAN Latent Space

detr. End-to-End Object Detection with Transformers

stylegan2-distillation. Python

DeepPruner. DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)

multiperson. Code repository for the paper: "Coherent Reconstruction of Multiple Humans from a Single Image" in CVPR'20

face-parsing.PyTorch. Using modified BiSeNet for face parsing in PyTorch

torchcv. TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision

HTML4Vision. A simple HTML visualization tool for computer vision research :hammer_and_wrench:

data-efficient-gans. [NeurIPS 2020] Differentiable Augmentation for Data-Efficient GAN Training

CNNDetection. Code for the paper: CNN-generated images are surprisingly easy to spot... for now https://peterwang512.github.io/CNNDetection/

cocostuff. The official homepage of the COCO-Stuff dataset.

PSPNet. Pyramid Scene Parsing Network, CVPR2017.

pytorch-deeplab-xception. DeepLab v3+ model in PyTorch. Support different backbones.

panoptic-deeplab. This is Pytorch re-implementation of our CVPR 2020 paper "Panoptic-DeepLab: A Simple, Strong, and Fast Baseline for Bottom-Up Panoptic Segmentation" (https://arxiv.org/abs/1911.10194)

sample-imageinpainting-HiFill. Python

mmdetection. OpenMMLab Detection Toolbox and Benchmark

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

PoseWarper. Learning Temporal Pose Estimation from Sparsely Labeled Videos

DiverseOutpaint. This is a Pytorch Implementation of Multimodal Image Outpainting

pytorch-cnn-visualizations. Pytorch implementation of convolutional neural network visualization techniques

deepcluster. Deep Clustering for Unsupervised Learning of Visual Features

synthetic-computer-vision. A list of synthetic dataset and tools for computer vision

Super-SloMo. PyTorch implementation of Super SloMo by Jiang et al.

SRGAN. A PyTorch implementation of SRGAN based on CVPR 2017 paper "Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network"

wgan-gp. A pytorch implementation of Paper "Improved Training of Wasserstein GANs"

pggan-pytorch. :fire::fire: PyTorch implementation of "Progressive growing of GANs (PGGAN)" :fire::fire:

mAP. mean Average Precision - This code evaluates the performance of your neural net for object recognition.

Detectron.pytorch. A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

capsule-networks. A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".