Cong Wang's repositories
FD-GAN
FD-GAN: Generative adversarial Networks with Fusion-discriminator for Single Image Dehazing(AAAI'20)
halonet-pytorch
Implementation of the 😇 Attention layer from the paper, Scaling Local Self-Attention For Parameter Efficient Visual Backbones
image-restoration-sde
Official PyTorch implementation for "Image Restoration with Mean-Reverting Stochastic Differential Equations".
ImageProcessingTransformer
Third party Pytorch implement of Image Processing Transformer (Pre-Trained Image Processing Transformer arXiv:2012.00364v2)
IQA-PyTorch
Image Quality Assessment Toolbox with PyTorch, including LPIPS, FID, NIQE, NRQM(Ma), MUSIQ, NIMA, DBCNN, WaDIQaM, BRISQUE, PI and more...
low-light-image-enhancement-resources
This is a resouce list for low light image enhancement
mae
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Multicore-TSNE
Parallel t-SNE implementation with Python and Torch wrappers.
Poisson_flow
Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models"
pytorch-fid
Compute FID scores with PyTorch.
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
Restormer
Official repository for "Restormer: Efficient Transformer for High-Resolution Image Restoration". SOTA for motion deblurring, image deraining, denoising (Gaussian/real data), and defocus deblurring.
RUAS
this is the official code for the paper "Retinex-inspired Unrolling with Cooperative Prior Architecture Search for Low-light Image Enhancement"
Saliency-Evaluation-Toolbox
This Toolbox contains E-measure, S-measure, weighted F & F-measure, MAE and PR curves or bar metrics for salient object detection.
SCGAN
The implementation of 'Image synthesis via semantic composition', ICCV2021.
SDWNet
[ICCVW2021] An Straight Dilated Network with Wavelet for image Deblurring
TransGAN
[NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang
U-Mamba
U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation
VR-Baseline
Video Restoration Toolbox including FGST (ICML 2022), S2SVR (ICML 2022), etc.
WSR
Official PyTorch implementation of WSR (Towards Lighter and Faster: Learning Wavelets Progressively for Image Super-Resolution) accepted by ACMMM2020.