swin-transformer

OpenMMLab Detection Toolbox and Benchmark

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

Implementation of popular deep learning networks with TensorRT network definition API

A treasure chest for visual classification and recognition powered by PaddlePaddle

OpenMMLab Pre-training Toolbox and Benchmark

Integrate deep learning models for image classification | Backbone learning/comparison/magic modification project

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.

This is an official implementation for "SimMIM: A Simple Framework for Masked Image Modeling".

More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and (cbam，dcn and so on), and tensorrt

This is an official implementation for "Self-Supervised Learning with Swin Transformers".

TransMorph: Transformer for Unsupervised Medical Image Registration (PyTorch)

PASSL包含 SimCLR，MoCo v1/v2，BYOL，CLIP，PixPro，simsiam, SwAV, BEiT，MAE 等图像自监督算法以及 Vision Transformer，DEiT，Swin Transformer，CvT，T2T-ViT，MLP-Mixer，XCiT，ConvNeXt，PVTv2 等基础视觉算法

Video Swin Transformer - PyTorch

Official Codes for "Uniform Masking: Enabling MAE Pre-training for Pyramid-based Vision Transformers with Locality"

SUNet: Swin Transformer with UNet for Image Denoising

Pytorch implementation for "A Novel Plug-in Module for Fine-Grained Visual Classification". fine-grained visual classification task.

PyTorch reimplementation of the paper "Swin Transformer V2: Scaling Up Capacity and Resolution" (CVPR 2022)

GRIT: Faster and Better Image-captioning Transformer (ECCV 2022)

Paddle Large Scale Classification Tools，supports ArcFace, CosFace, PartialFC, Data Parallel + Model Parallel. Model includes ResNet, ViT, Swin, DeiT, CaiT, FaceViT, MoCo, MAE, ConvMAE, CAE.