swin-transformer · GitHub Topics

OpenMMLab Detection Toolbox and Benchmark

object-detection instance-segmentation fast-rcnn faster-rcnn mask-rcnn cascade-rcnn ssd retinanet PyTorch panoptic-segmentation rtmdet swin-transformer transformer vision-transformer yolo convnext detr grounding-dino

Python 31.17 k

10 个月前

microsoft / Swin-Transformer

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

swin-transformer image-classification object-detection semantic-segmentation imagenet mscoco ade20k mask-rcnn

Python 14.89 k

1 年前

open-mmlab / mmsegmentation

MMSegmentation 是一个基于 PyTorch 的语义分割开源工具箱。它是 OpenMMLab 项目的一部分。

semantic-segmentation PyTorch pspnet deeplabv3 transformer swin-transformer realtime-segmentation vessel-segmentation retinal-vessel-segmentation image-segmentation medical-image-segmentation

Python 8.97 k

10 个月前

wang-xinyu / tensorrtx

Implementation of popular deep learning networks with TensorRT network definition API

tensorrt mnasnet resnet squeezenet yolov3 mobilenetv2 mobilenetv3 yolov3-spp retinaface yolov4 arcface yolov5 crnn swin-transformer detr yolov7 yolov8 yolov9 yolo11

C++ 7.39 k

1 个月前

PaddlePaddle / PaddleClas

PaddleClas 是一个为工业界和学术界所准备的图像识别任务工具集

image-classification knowledge-distillation autoaugment randaugment gridmask cutmix deit repvgg swin-transformer pretrained-models image-recognition

Python 5.67 k

1 个月前

open-mmlab / mmpretrain

#计算机科学#OpenMMLab Pre-training Toolbox and Benchmark

image-classification resnet mobilenet PyTorch 深度学习 swin-transformer beit clip constrastive-learning convnext masked-image-modeling moco pretrained-models self-supervised-learning vision-transformer multimodal

Python 3.68 k

7 个月前

Fafa-DL / Awesome-Backbones

#计算机科学#Integrate deep learning models for image classification | Backbone learning/comparison/magic modification project

PyTorch image-classification transformer cnn 深度学习 resnet swin-transformer

Python 1.78 k

5 个月前

Gumpest / YOLOv5-Multibackbone-Compression

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantizat...

pplcnet swin-transformer cbam ncnn tensorrt

Python 1.01 k

3 年前

microsoft / SimMIM

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

self-supervised-learning masked-image-modeling image-classification swin-transformer

Python 977

3 年前

Bobo-y / flexible-yolov5

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

yolov5 resnet Backbone.js cbam PyTorch shufflenet hrnet tensorrt object-detection swin-transformer gcn yolov3 sparsity

Python 675

10 个月前

SwinTransformer / Transformer-SSL

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

transformer swin-transformer self-supervised-learning

Python 657

4 年前

junyuchen245 / TransMorph_Transformer_for_Medical_Image_Registration

#计算机科学#TransMorph: Transformer for Unsupervised Medical Image Registration (PyTorch)

transformer vision-transformer image-registration image-alignment 深度学习 diffeomorphism swin-transformer

Python 522

24 天前

PaddlePaddle / PASSL

#计算机科学#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 等基础视觉算法

深度学习 moco simclr clip self-supervised-learning paddle swin-transformer vision-transformer beit convnext vit deit pvt swav

Python 283

2 年前

haofanwang / video-swin-transformer-pytorch

Video Swin Transformer - PyTorch

swin-transformer

Python 254

3 年前

implus / UM-MAE

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

ade20k coco imagenet-classification masked-image-modeling self-supervised-learning swin-transformer

Jupyter Notebook 242

3 年前

FanChiMao / SUNet

SUNet: Swin Transformer with UNet for Image Denoising

unet swin-transformer image-denoising

Python 236

2 年前

chou141253 / FGVC-PIM

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

swin-transformer efficientnet resnet vision-transformer

Python 200

2 年前

ChristophReich1996 / Swin-Transformer-V2

#计算机科学#PyTorch reimplementation of the paper "Swin Transformer V2: Scaling Up Capacity and Resolution" (CVPR 2022)

机器视觉 transformer attention swin-transformer 深度学习 PyTorch vision-transformer

Python 197

3 年前

davidnvq / grit

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

detr eccv2022 swin-transformer transformer-models image-captioning object-detection

Python 192

2 年前

PaddlePaddle / PLSC

#人脸识别#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.

face-recognition arcface cosface large-scale paddlepaddle paddle distributed-training deit resnet vit swin-transformer

Python 152

2 年前