swin.md 4.89 KB
Newer Older
novice's avatar
novice committed
1
2
3
4
5
6
7
8
9
10
<!--Copyright 2022 The HuggingFace Team. All rights reserved.

Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
the License. You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
specific language governing permissions and limitations under the License.
11
12
13
14

鈿狅笍 Note that this file is in Markdown but contain specific syntax for our doc-builder (similar to MDX) that may not be
rendered properly in your Markdown viewer.

novice's avatar
novice committed
15
16
17
18
19
20
-->

# Swin Transformer

## Overview

21
22
The Swin Transformer was proposed in [Swin Transformer: Hierarchical Vision Transformer using Shifted Windows](https://arxiv.org/abs/2103.14030)
by Ze Liu, Yutong Lin, Yue Cao, Han Hu, Yixuan Wei, Zheng Zhang, Stephen Lin, Baining Guo.
novice's avatar
novice committed
23
24
25

The abstract from the paper is the following:

26
27
28
29
30
31
32
33
34
35
36
*This paper presents a new vision Transformer, called Swin Transformer, that capably serves as a general-purpose backbone
for computer vision. Challenges in adapting Transformer from language to vision arise from differences between the two domains,
such as large variations in the scale of visual entities and the high resolution of pixels in images compared to words in text.
To address these differences, we propose a hierarchical Transformer whose representation is computed with \bold{S}hifted
\bold{win}dows. The shifted windowing scheme brings greater efficiency by limiting self-attention computation to non-overlapping
local windows while also allowing for cross-window connection. This hierarchical architecture has the flexibility to model at
various scales and has linear computational complexity with respect to image size. These qualities of Swin Transformer make it
compatible with a broad range of vision tasks, including image classification (87.3 top-1 accuracy on ImageNet-1K) and dense
prediction tasks such as object detection (58.7 box AP and 51.1 mask AP on COCO test-dev) and semantic segmentation
(53.5 mIoU on ADE20K val). Its performance surpasses the previous state-of-the-art by a large margin of +2.7 box AP and
+2.6 mask AP on COCO, and +3.2 mIoU on ADE20K, demonstrating the potential of Transformer-based models as vision backbones.
novice's avatar
novice committed
37
38
39
The hierarchical design and the shifted window approach also prove beneficial for all-MLP architectures.*

<img src="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/swin_transformer_architecture.png"
40
alt="drawing" width="600"/>
novice's avatar
novice committed
41
42
43

<small> Swin Transformer architecture. Taken from the <a href="https://arxiv.org/abs/2102.03334">original paper</a>.</small>

44
This model was contributed by [novice03](https://huggingface.co/novice03). The Tensorflow version of this model was contributed by [amyeroberts](https://huggingface.co/amyeroberts). The original code can be found [here](https://github.com/microsoft/Swin-Transformer).
novice's avatar
novice committed
45

46
47
48
49
## Usage tips

- Swin pads the inputs supporting any input height and width (if divisible by `32`).
- Swin can be used as a *backbone*. When `output_hidden_states = True`, it will output both `hidden_states` and `reshaped_hidden_states`. The `reshaped_hidden_states` have a shape of `(batch, num_channels, height, width)` rather than `(batch_size, sequence_length, num_channels)`.
novice's avatar
novice committed
50

NielsRogge's avatar
NielsRogge committed
51
52
53
54
55
56
57
## Resources

A list of official Hugging Face and community (indicated by 馃寧) resources to help you get started with Swin Transformer.

<PipelineTag pipeline="image-classification"/>

- [`SwinForImageClassification`] is supported by this [example script](https://github.com/huggingface/transformers/tree/main/examples/pytorch/image-classification) and [notebook](https://colab.research.google.com/github/huggingface/notebooks/blob/main/examples/image_classification.ipynb).
58
- See also: [Image classification task guide](../tasks/image_classification)
NielsRogge's avatar
NielsRogge committed
59
60
61
62
63
64
65

Besides that:

- [`SwinForMaskedImageModeling`] is supported by this [example script](https://github.com/huggingface/transformers/tree/main/examples/pytorch/image-pretraining).

If you're interested in submitting a resource to be included here, please feel free to open a Pull Request and we'll review it! The resource should ideally demonstrate something new instead of duplicating an existing resource.

novice's avatar
novice committed
66
67
68
69
## SwinConfig

[[autodoc]] SwinConfig

70
71
<frameworkcontent>
<pt>
novice's avatar
novice committed
72
73
74
75
76
77

## SwinModel

[[autodoc]] SwinModel
    - forward

NielsRogge's avatar
NielsRogge committed
78
79
80
81
## SwinForMaskedImageModeling

[[autodoc]] SwinForMaskedImageModeling
    - forward
novice's avatar
novice committed
82

NielsRogge's avatar
NielsRogge committed
83
84
## SwinForImageClassification

novice's avatar
novice committed
85
[[autodoc]] transformers.SwinForImageClassification
86
87
    - forward

88
89
90
</pt>
<tf>

91
92
93
94
95
96
97
98
99
100
101
102
103
104
## TFSwinModel

[[autodoc]] TFSwinModel
    - call

## TFSwinForMaskedImageModeling

[[autodoc]] TFSwinForMaskedImageModeling
    - call

## TFSwinForImageClassification

[[autodoc]] transformers.TFSwinForImageClassification
    - call
105
106
107

</tf>
</frameworkcontent>