Flops counter for neural networks in pytorch framework

An optimizer that trains as fast as Adam and as good as SGD.

A PyTorch Library for Meta-learning Research

SNIPER / AutoFocus is an efficient multi-scale object detection training / inference algorithm

Benchmarks for popular CNN models

Convolutional recurrent network in pytorch

An intuitive library to add plotting functionality to scikit-learn objects.

Code for the paper "Improved Techniques for Training GANs"

micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantization and Training of Neural Networks for Efficient Integer-…

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

Codebase for Image Classification Research, written in PyTorch.

[ICCV 2019] TSM: Temporal Shift Module for Efficient Video Understanding

Prompt Learning for Vision-Language Models (IJCV'22, CVPR'22)

Caffe implementation of Google MobileNet SSD detection network, with pretrained weights on VOC0712 and mAP=0.727.

A CV toolkit for my papers.

Pytorch implementation of cnn network

单阶段通用目标检测器

plotting in the terminal

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Object Detection and Instance Segmentation.

Codes for our paper "CenterNet: Keypoint Triplets for Object Detection" .

An open-source toolbox for action understanding based on PyTorch

mobilenetv3 with pytorch，provide pre-train model

Classification with PyTorch.

PyTorch for Semantic Segmentation

Pytorch implementation of various Knowledge Distillation (KD) methods.

在 oxford hand 数据集上对 YOLOv3 做模型剪枝（network slimming）

A quickstart and benchmark for pytorch distributed training.

Official Implementation of 'Fast AutoAugment' in PyTorch.