ZhendongWang6

Zhendong Wang's starred repositories

TransUNet

This repository includes the official project of TransUNet, presented in our paper: TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation.

Language:PythonApache-2.0225900

NBNet

Pytorch UNOFFICIAL implement "NBNet: Noise Basis Learning for Image Denoising with Subspace Projection"

Language:Python2400

DPT

Dense Prediction Transformers

Language:PythonMIT191200

VIDIT

(CVPR21/ECCV20 Workshops) Official repo: Virtual Image Dataset for Illumination Transfer (VIDIT)

MIT15400

awesome-NeRF

A curated list of awesome neural radiance fields papers

Language:TeXMIT635400

imgaug

Image augmentation for machine learning experiments.

Language:PythonMIT1427400

SimCLR

A PyTorch implementation of SimCLR based on ICML 2020 paper "A Simple Framework for Contrastive Learning of Visual Representations"

Language:Python52800

UnprocessDenoising_PyTorch

Unofficial PyTorch code for the paper - Unprocessing Images for Learned Raw Denoising, CVPR'19

Language:Python16900

mean-teacher

A state-of-the-art semi-supervised method for image recognition

Language:PythonNOASSERTION156100

semi-supervised-pytorch

Implementations of various VAE-based semi-supervised and generative models in PyTorch

Language:PythonMIT70400

simclr

SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners

Language:Jupyter NotebookApache-2.0401300

pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

Language:PythonMIT478100

T2T-ViT

ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet

Language:Jupyter NotebookNOASSERTION113100

NBNet_Pytorch

ReImplement of 'NBNet: Noise Basis Learning for Image Denoising with Subspace Projection' in Pytorch

Language:Python1900

MSBDN-DFF

The source code of CVPR 2020 paper "Multi-Scale Boosted Dehazing Network with Dense Feature Fusion"

Language:Python32900

pixel-level-contrastive-learning

Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch

Language:PythonMIT24900

byol-pytorch

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch

Language:PythonMIT171600

PReNet

Progressive Image Deraining Networks: A Better and Simpler Baseline (CVPR 2019)

Language:Python22900

pytorch-vit

An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale

Language:PythonMIT28100

deit

Official DeiT repository

Language:PythonApache-2.0395000

SAN

Exploring Self-attention for Image Recognition, CVPR2020.

Language:PythonMIT74800

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also support StyleGAN2, DFDNet.

Language:PythonApache-2.0650700

KAIR

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR

Language:PythonMIT281200

COCO-GAN

COCO-GAN: Generation by Parts via Conditional Coordinating (ICCV 2019 oral)

Language:PythonMIT26800

Total_Variation_Loss.pytorch

pytorch implementation of tv loss

Language:Python10000

DnCNN

Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising (TIP, 2017)

Language:MATLAB141100

PSConv

[ECCV 2020] PSConv: Squeezing Feature Pyramid into One Compact Poly-Scale Convolutional Layer

Language:PythonMIT17500

super-resolution

Language:Python1200

longformer

Longformer: The Long-Document Transformer

Language:PythonApache-2.0200100