Add Swin-T OBB config (#35)

Y-T-G · Jan 20, 2025 · f685f64 · f685f64
1 parent b4d429f
commit f685f64
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/cfg/obb/swin_t.yaml b/cfg/obb/swin_t.yaml
@@ -0,0 +1,23 @@
+# Metadata
+author: Y-T-G
+task: obb
+keywords: [transformer, swin]
+description: Swin-T pretrained backbone with OBB head.
+flops: 67.0
+parameters: 32380000
+min_version: 8.3.64
+
+# Model
+nc: 1
+strides: [8, 16, 32]
+backbone:
+  # [from, repeats, module, args]
+  - [-1, 1, TorchVision, [768, "swin_t", "DEFAULT", True, 5, True]]
+  - [0, 1, Index, [192, 4]]               # (1, 80, 80, 192) - 1
+  - [-1, 1, torchvision.ops.Permute, [[0, 3, 1, 2]]]  # (1, 192, 80, 80) - 2
+  - [0, 1, Index, [384, 6]]               # (1, 40, 40, 384) - 3
+  - [-1, 1, torchvision.ops.Permute, [[0, 3, 1, 2]]]  # (1, 384, 40, 40) - 4
+  - [0, 1, Index, [768, 8]]               # (1, 20, 20, 768) - 5
+  - [-1, 1, torchvision.ops.Permute, [[0, 3, 1, 2]]]  # (1, 768, 20, 20) - 6
+head:
+  - [[2, 4, 6], 1, OBB, [nc, 1]]