He Zhang's starred repositories

Bringing-Old-Photos-Back-to-Life

Bringing Old Photo Back to Life (CVPR 2020 oral)

Language:PythonLicense:MITStargazers:14575Issues:285Issues:248

CodeFormer

[NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer

Language:PythonLicense:NOASSERTIONStargazers:13887Issues:285Issues:319

Dreambooth-Stable-Diffusion

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

Language:Jupyter NotebookLicense:MITStargazers:7490Issues:92Issues:146

lama

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

Language:Jupyter NotebookLicense:Apache-2.0Stargazers:7410Issues:80Issues:242

stylegan3

Official PyTorch implementation of StyleGAN3

Language:PythonLicense:NOASSERTIONStargazers:6235Issues:56Issues:252

ConvNeXt

Code release for ConvNeXt model

Language:PythonLicense:MITStargazers:5597Issues:33Issues:130

SwinIR

SwinIR: Image Restoration Using Swin Transformer (official repository)

Language:PythonLicense:Apache-2.0Stargazers:4168Issues:55Issues:139

imaginaire

NVIDIA's Deep Imagination Team's PyTorch Library

Language:PythonLicense:NOASSERTIONStargazers:3958Issues:109Issues:171

vispy

Main repository for Vispy

Language:PythonLicense:NOASSERTIONStargazers:3239Issues:116Issues:1426

ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)

Language:PythonLicense:Apache-2.0Stargazers:2775Issues:20Issues:270

PixArt-alpha

PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis

Language:PythonLicense:AGPL-3.0Stargazers:2385Issues:41Issues:0

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Language:PythonLicense:BSD-3-ClauseStargazers:2376Issues:69Issues:1404

awesome-implicit-representations

A curated list of resources on implicit neural representations.

License:MITStargazers:2376Issues:113Issues:0

SegFormer

Official PyTorch implementation of SegFormer

Language:PythonLicense:NOASSERTIONStargazers:2322Issues:31Issues:142

PVT

Official implementation of PVT series

Language:PythonLicense:Apache-2.0Stargazers:1664Issues:22Issues:108

alias-free-gan

Alias-Free GAN project website and code

MaskFormer

Per-Pixel Classification is Not All You Need for Semantic Segmentation (NeurIPS 2021, spotlight)

Language:PythonLicense:NOASSERTIONStargazers:1305Issues:25Issues:69

Barbershop

Barbershop: GAN-based Image Compositing using Segmentation Masks (SIGGRAPH Asia 2021)

Language:PythonLicense:MITStargazers:1285Issues:65Issues:64

co-mod-gan

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

Language:PythonLicense:NOASSERTIONStargazers:438Issues:15Issues:61

Awesome-Image-Harmonization

A curated list of papers, code and resources pertaining to image harmonization.

CoCosNet-v2

CoCosNet v2: Full-Resolution Correspondence Learning for Image Translation

Language:PythonLicense:MITStargazers:335Issues:18Issues:27

segformer-pytorch

Implementation of Segformer, Attention + MLP neural network for segmentation, in Pytorch

Language:PythonLicense:MITStargazers:318Issues:9Issues:12

image_harmonization

[WACV2021] Foreground-aware Semantic Representations for Image Harmonization https://arxiv.org/abs/2006.00809

Language:PythonLicense:MPL-2.0Stargazers:269Issues:16Issues:10

ViTAE-Transformer-Matting

A comprehensive list [AIM@IJCAI'21, P3M@MM'21, GFM@IJCV'22, RIM@CVPR'23, P3MNet@IJCV'23] of our research works related to image matting, including papers, codes, datasets, demos, and citations. Note: The repo for [IJCV'23] "Rethinking Portrait Matting with Privacy Preserving" has been moved to: https://github.com/ViTAE-Transformer/P3M-Net

Language:TeXLicense:MITStargazers:227Issues:6Issues:6

hyperseg

HyperSeg - Official PyTorch Implementation

Language:PythonLicense:CC0-1.0Stargazers:205Issues:7Issues:36

LVT

Lite Vision Transformer (CVPR 2022)

Language:PythonLicense:NOASSERTIONStargazers:131Issues:3Issues:6

SSHarmonization

[ICCV'2021] "SSH: A Self-Supervised Framework for Image Harmonization", Yifan Jiang, He Zhang, Jianming Zhang, Yilin Wang, Zhe Lin, Kalyan Sunkavalli, Simon Chen, Sohrab Amirghodsi, Sarah Kong, Zhangyang Wang

Language:Jupyter NotebookLicense:NOASSERTIONStargazers:97Issues:9Issues:11

skylibs

Python library to handle 360° environment maps

Language:PythonLicense:LGPL-3.0Stargazers:82Issues:5Issues:12

SMDDP-Examples

Repository for Forked Deep Learning Examples Used in SageMaker Data Distributed Training

Language:PythonStargazers:9Issues:7Issues:0