Wenjie Zhang's repositories
PaddleSlim
PaddleSlim is an open-source library for deep model compression and architecture search.
maxim-pytorch
[CVPR 2022 Oral] PyTorch re-implementation for "MAXIM: Multi-Axis MLP for Image Processing", with *training code*. Official Jax repo: https://github.com/google-research/maxim
nni
An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
OpenXR
OpenXR: Open Source Grayscale Image Processing Toolbox
mmdeploy
OpenMMLab Model Deployment Framework
instant-ngp
Instant neural graphics primitives: lightning fast NeRF and more
torch-ngp
A pytorch CUDA extension implementation of instant-ngp (sdf and nerf), with a GUI.
openvino
OpenVINO™ Toolkit repository
ngp_pl
Instant-ngp in pytorch+cuda trained with pytorch-lightning (high quality with high speed, with only few lines of legible code)
nerf_pl
NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning
Restormer
[CVPR 2022--Oral] Restormer: Efficient Transformer for High-Resolution Image Restoration. SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.
RePaint
Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022
YOLOX
YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/
SUNet
SUNet: Swin Transformer with UNet for Image Denoising
TTSR
[CVPR'20] TTSR: Learning Texture Transformer Network for Image Super-Resolution
yolact
A simple, fully convolutional model for real-time instance segmentation.
SinNeRF
[ECCV 2022] "SinNeRF: Training Neural Radiance Fields on Complex Scenes from a Single Image", Dejia Xu, Yifan Jiang, Peihao Wang, Zhiwen Fan, Humphrey Shi, Zhangyang Wang
DeblurGANv2
[ICCV 2019] "DeblurGAN-v2: Deblurring (Orders-of-Magnitude) Faster and Better" by Orest Kupyn, Tetiana Martyniuk, Junru Wu, Zhangyang Wang
yolov7
Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
YOLOv6
YOLOv6: a single-stage object detection framework dedicated to industrial applications.
oidn
Intel® Open Image Denoise library
RCAN
PyTorch code for our ECCV 2018 paper "Image Super-Resolution Using Very Deep Residual Channel Attention Networks"
Efficient-Segmentation-Networks
Lightweight models for real-time semantic segmentationon PyTorch (include SQNet, LinkNet, SegNet, UNet, ENet, ERFNet, EDANet, ESPNet, ESPNetv2, LEDNet, ESNet, FSSNet, CGNet, DABNet, Fast-SCNN, ContextNet, FPENet, etc.)
FasterSeg
[ICLR 2020] "FasterSeg: Searching for Faster Real-time Semantic Segmentation" by Wuyang Chen, Xinyu Gong, Xianming Liu, Qian Zhang, Yuan Li, Zhangyang Wang
RobustNet
Official PyTorch implementation of RobustNet (CVPR 2021 Oral)
RLFN
Winner of runtime track in NTIRE 2022 challenge on Efficient Super-Resolution