DerainZoo for collecting deraining methods, datasets, and codes.

PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"

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

Multi-Scale Progressive Fusion Network for Single Image Deraining

Dual Attention Residual Group Networks for Single Image Deraining

Syn2Real Transfer Learning for Image Deraining using Gaussian Processes

Count the MACs / FLOPs of your PyTorch model.

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

Stochastic Bundle Adjustment for Efficient and Scalable Structure from Motion (ECCV 2020)

Implementation of Convolutional LSTM in PyTorch.

Official repository for the paper F, B, Alpha Matting

Getting to 99% Accuracy in Interactive Segmentation and Interactive Training and Architecture for Deep Object Selection

Latex code for making neural networks diagrams

Pytorch implementation for image compression and reconstruction via autoencoder

The codes of Layered Conceptual Image Compression via Semantic Synthesis

PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.

Rethinking the Value of Network Pruning (Pytorch) (ICLR 2019)

Real-time dense visual SLAM system

PyTorch code for "Pyramid Attention Networks for Image Restoration" with new SOTA results on multiple image restoration tasks: denoising, demosaicing, compression artifact reduction, super-resolution

A curated list of neural network pruning resources.

Learning Convolutional Neural Networks with Interactive Visualization. https://poloclub.github.io/cnn-explainer/

A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.

Yolo v4 (v3/v2) - Windows and Linux version of Darknet Neural Networks for object detection (Tensor Cores are used)

Repository of the paper "Learned Image Compression with Discretized Gaussian Mixture Likelihoods and Attention Modules"

MVSNet (ECCV2018) & R-MVSNet (CVPR2019)

Project page for paper A Late Fusion CNN for Digital Matting.

PyTorch implementation of SENet

PyTorch Implementation of "Lossless Image Compression through Super-Resolution"