zllrunning / awesome-pytorch-toolkit

A list of pytorch toolkits for CV application.

awesome-pytorch-toolkit

A list of pytorch toolkits for CV application.

Contents

• Tutorials for PyTorch
• Classification
• Object detection
• Semantic segmentation
• GAN
• Face
• Pose estimation
• Object tracking
• Person Reid
• Image super-resolution
• Action understanding
• Crowd counting
• Image inpainting
• Augmentation
• Tools

Tutorials for PyTorch

• Official Pytorch Tutorial : DEEP LEARNING WITH PYTORCH: A 60 MINUTE BLITZ

• yunjey / pytorch-tutorial : PyTorch Tutorial for Deep Learning Researchers

• sgrvinod / Deep-Tutorials-for-PyTorch : In-depth tutorials for implementing deep learning models on your own with PyTorch

• zergtant / pytorch-handbook : pytorch handbook is an open source book

Classification

• kuangliu / pytorch-cifar : Train CIFAR10 with PyTorch

• Cadene / pretrained-models.pytorch : Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc

• rwightman / pytorch-image-models : PyTorch image models, scripts, pretrained weights -- (SE)ResNet/ResNeXT, DPN, EfficientNet, MobileNet-V3/V2/V1, MNASNet, Single-Path NAS, FBNet, and more

Object detection

• mmdetection : Open MMLab Detection Toolbox with PyTorch

• facebookresearch / maskrcnn-benchmark : Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch

• roytseng-tw / Detectron.pytorch : A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available

Semantic segmentation

• CSAILVision/ semantic-segmentation-pytorch : Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset

• meetshah1995 / pytorch-semseg : Semantic Segmentation Architectures Implemented in PyTorch

• hszhao / semseg : Semantic Segmentation in Pytorch

• Tramac / awesome-semantic-segmentation-pytorch : Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet)

• ycszen / TorchSeg : Fast, modular reference implementation and easy training of Semantic Segmentation algorithms in PyTorch

• donnyyou / torchcv-seg : Open Source for Semantic Segmentation

• dbolya / yolact : A simple, fully convolutional model for real-time instance segmentation

• facebookresearch / maskrcnn-benchmark : Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch

GAN

• eriklindernoren / PyTorch-GAN : PyTorch implementations of Generative Adversarial Networks

• znxlwm / pytorch-generative-model-collections : Collection of generative models in Pytorch version

• junyanz /pytorch-CycleGAN-and-pix2pix : Image-to-image translation in PyTorch (e.g., horse2zebra, edges2cats, and more)

Face

• ZhaoJ9014 / face.evoLVe.PyTorch ： High-Performance Face Recognition Library on PyTorch

• 1adrianb / face-alignment : 2D and 3D Face alignment library build using pytorch

• cleardusk / 3DDFA : Face Alignment in Full Pose Range: A 3D Total Solution

• HRNet / HRNet-Facial-Landmark-Detection : High-resolution representation learning (HRNets) for facial landmark detection

• wujiyang / Face_Pytorch : face recognition algorithms in pytorch framework, including arcface, cosface, sphereface and so on

Pose estimation

• microsoft / human-pose-estimation.pytorch : Simple Baselines for Human Pose Estimation and Tracking

• MVIG-SJTU / AlphaPose : Real-Time and Accurate Multi-Person Pose Estimation&Tracking System

• bearpaw / pytorch-pose : A PyTorch toolkit for 2D Human Pose Estimation

• YuliangXiu / MobilePose-pytorch : Light-weight Single Person Pose Estimator

Object tracking

• STVIR / pysot : SenseTime Research platform for single object tracking, implementing algorithms like SiamRPN and SiamMask

• visionml / pytracking : Visual tracking library based on PyTorch

• zllrunning / SiameseX.PyTorch : A simplified PyTorch implementation of Siamese networks for tracking: SiamFC, SiamVGG, SiamDW, SiamRPN, SiamRPN-VGG

• researchmm / SiamDW : Deeper and Wider Siamese Networks for Real-Time Visual Tracking

Person Reid

• KaiyangZhou / deep-person-reid : Deep learning person re-identification in PyTorch

• layumi / Person_reID_baseline_pytorch : A tiny, friendly, strong pytorch implement of person re-identification baseline

• michuanhaohao / reid-strong-baseline : Bag of Tricks and A Strong Baseline for Deep Person Re-identification

Image super-resolution

• thstkdgus35 / EDSR-PyTorch : PyTorch implementation of EDSR, MDSR, DBPN, RDN, RCAN

• xinntao / BasicSR : Basic Super-Resolution codes for development. Includes ESRGAN, SFT-GAN for training and testing

Action understanding

• open-mmlab / mmaction : MMAction is an open source toolbox for action understanding based on PyTorch. It is a part of the open-mmlab project developed by Multimedia Laboratory, CUHK.

Crowd counting

• gjy3035 / C-3-Framework : An open-source PyTorch code for crowd counting

Image inpainting

• NVIDIA / partialconv : This is the PyTorch implementation of partial convolution layer. It can serve as a new padding scheme; it can also be used for image inpainting

• knazeri / edge-connect : Generative Image Inpainting with Adversarial Edge Learning

• lyndonzheng / Pluralistic-Inpainting : Pluralistic Image Completion

Augmentation

• aleju / imgaug : Image augmentation for machine learning experiments

• albu / albumentations : fast image augmentation library and easy to use wrapper around other libraries

• mdbloice / Augmentor : Image augmentation library in Python for machine learning

• torchvideo / torchvideo : Datasets, transforms and samplers for video in PyTorch

• hassony2 / torch_videovision : Transforms for video datasets in pytorch

Tools

• facebookresearch / visdom : A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy

• zasdfgbnm / TorchSnooper : Debug PyTorch code using PySnooper

• sovrasov / flops-counter.pytorch : Flops counter for convolutional networks in pytorch framework

• Swall0w / torchstat : Model analyzer in PyTorch

Contributing

Your contributions are always welcome! To add, remove, or change things on the list: Submit a pull request.