gitea/Swin-Transformer
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Initial commit

Browse Source
This commit is contained in:
v-zeliu1
2021-04-13 00:34:56 +08:00
parent ce5bae042d
commit 3dc2a55301
24 changed files with 2482 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
configs/swin_base_patch4_window12_384.yaml Normal file
View File

@@ -0,0 +1,13 @@
# only for evaluation
DATA:
IMG_SIZE: 384
MODEL:
TYPE: swin
NAME: swin_base_patch4_window12_384
SWIN:
EMBED_DIM: 128
DEPTHS: [ 2, 2, 18, 2 ]
NUM_HEADS: [ 4, 8, 16, 32 ]
WINDOW_SIZE: 12
TEST:
CROP: False

9
configs/swin_base_patch4_window7_224.yaml Normal file
View File

@@ -0,0 +1,9 @@
MODEL:
TYPE: swin
NAME: swin_base_patch4_window7_224
DROP_PATH_RATE: 0.5
SWIN:
EMBED_DIM: 128
DEPTHS: [ 2, 2, 18, 2 ]
NUM_HEADS: [ 4, 8, 16, 32 ]
WINDOW_SIZE: 7

13
configs/swin_large_patch4_window12_384.yaml Normal file
View File

@@ -0,0 +1,13 @@
# only for evaluation
DATA:
IMG_SIZE: 384
MODEL:
TYPE: swin
NAME: swin_large_patch4_window12_384
SWIN:
EMBED_DIM: 192
DEPTHS: [ 2, 2, 18, 2 ]
NUM_HEADS: [ 6, 12, 24, 48 ]
WINDOW_SIZE: 12
TEST:
CROP: False

9
configs/swin_large_patch4_window7_224.yaml Normal file
View File

@@ -0,0 +1,9 @@
# only for evaluation
MODEL:
TYPE: swin
NAME: swin_large_patch4_window7_224
SWIN:
EMBED_DIM: 192
DEPTHS: [ 2, 2, 18, 2 ]
NUM_HEADS: [ 6, 12, 24, 48 ]
WINDOW_SIZE: 7

9
configs/swin_small_patch4_window7_224.yaml Normal file
View File

@@ -0,0 +1,9 @@
MODEL:
TYPE: swin
NAME: swin_small_patch4_window7_224
DROP_PATH_RATE: 0.3
SWIN:
EMBED_DIM: 96
DEPTHS: [ 2, 2, 18, 2 ]
NUM_HEADS: [ 3, 6, 12, 24 ]
WINDOW_SIZE: 7

9
configs/swin_tiny_patch4_window7_224.yaml Normal file
View File

@@ -0,0 +1,9 @@
MODEL:
TYPE: swin
NAME: swin_tiny_patch4_window7_224
DROP_PATH_RATE: 0.2
SWIN:
EMBED_DIM: 96
DEPTHS: [ 2, 2, 6, 2 ]
NUM_HEADS: [ 3, 6, 12, 24 ]
WINDOW_SIZE: 7