← all repositories

berniwal/swin-transformer-pytorch

PyTorch implementation of Swin Transformer, a hierarchical vision transformer for computer vision tasks.

swin-transformer-pytorch
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of the Swin Transformer architecture, a vision transformer designed for computer vision tasks. The model uses a shifted windowing scheme for self-attention to achieve linear computational complexity with respect to image size. It serves as a general-purpose backbone for tasks including image classification, object detection, and semantic segmentation, achieving strong performance on benchmarks such as ImageNet-1K, COCO, and ADE20K.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.