Count the MACs / FLOPs of your PyTorch model.

深度学习经典、新论文逐段精读

MulimgViewer is a multi-image viewer that can open multiple images in one interface, which is convenient for image comparison and image stitching.

SwinIR: Image Restoration Using Swin Transformer

Flops counter for convolutional networks in pytorch framework

Datasets, Transforms and Models specific to Computer Vision

Pytorch implementation of convolutional neural network visualization techniques

Awesome Incremental Learning

Latex code for making neural networks diagrams

DehazeZoo for collecting dehazing methods, datasets, and codes.

DerainZoo for collecting deraining methods, datasets, and codes.

《Pytorch模型训练实用教程》中配套代码

A CV toolkit for my papers.

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

Pytorch implementation of the paper: "A Unified Framework for Separating Superimposed Images", in CVPR 2020.

PyTorch implementations of Generative Adversarial Networks.

Exploring Self-attention for Image Recognition, CVPR2020.

Syn2Real Transfer Learning for Image Deraining using Gaussian Processes

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

Recurrent Squeeze-and-Excitation Context Aggregation Net for Single Image Deraining

Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"

experiments on Paper <Bag of Tricks for Image Classification with Convolutional Neural Networks> and other useful tricks to improve CNN acc

Implementation of "Gated Context Aggregation Network for Image Dehazing and Deraining"

Image Dehaze, Pytorch, An All-in-One Network for Dehazing, AOD-Net

Attentive Generative Adversarial Network for Raindrop Removal from A Single Image (CVPR 2018)

A PyTorch Implementation for Densely Connected Convolutional Networks (DenseNets)