Skip to content

Add Swin-T FPN and MobileNetv3 FPN configs #52

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Y-T-G merged 18 commits into from
Jan 21, 2025
Merged

Add Swin-T FPN and MobileNetv3 FPN configs #52

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
26de670
Add Swin-T Detect FPN config (#40)
Y-T-G Jan 21, 2025
120f93c
Add discussion link to detect/swin_t-fpn.yaml
actions-user Jan 21, 2025
71d6513
Reformat Swin-T config
Y-T-G Jan 21, 2025
7062916
Add Swin-T FPN Pose config (#42)
Y-T-G Jan 21, 2025
9e7acdd
Add Swin-T FPN Segment config (#43)
Y-T-G Jan 21, 2025
8c1f193
Add discussion link to segment/swin_t-fpn.yaml
actions-user Jan 21, 2025
1b9bb3f
Add output shape extractor (#44)
Y-T-G Jan 21, 2025
a3065c3
Add Swin-T FPN Pose config (#45)
Y-T-G Jan 21, 2025
5bdc641
Add discussion link to pose/swin_t-fpn.yaml
actions-user Jan 21, 2025
273a3cf
Add MobileNetv3-Large FPN Detect config (#49)
Y-T-G Jan 21, 2025
8e078f1
Add discussion link to detect/mobilenet_v3_large-fpn.yaml
actions-user Jan 21, 2025
05e6ce8
Add MobileNetv3-Small FPN Detect config (#50)
Y-T-G Jan 21, 2025
92b39e2
Add discussion link to detect/mobilenet_v3_small-fpn.yaml
actions-user Jan 21, 2025
68dbd50
Remove redundant quotes
Y-T-G Jan 21, 2025
575d522
Print error traceback for failed validation
Y-T-G Jan 21, 2025
5d26fd6
Fix PR validation script
Y-T-G Jan 21, 2025
a3b9891
Disable yaml validation on staging
Y-T-G Jan 21, 2025
90406c2
Disable YAML validation in staging
Y-T-G Jan 21, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .github/workflows/pr.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ on:

jobs:
validate-yaml:
if: github.head_ref != 'staging'
runs-on: ubuntu-latest
permissions:
pull-requests: write
Expand Down
2 changes: 1 addition & 1 deletion cfg/classify/convnext_tiny.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "convnext_tiny", "DEFAULT", True, 2]] # (1, 768, 20, 20) - 0
- [-1, 1, TorchVision, [768, convnext_tiny, DEFAULT, True, 2]] # (1, 768, 20, 20) - 0

head:
- [-1, 1, Classify, [nc]]
2 changes: 1 addition & 1 deletion cfg/classify/efficientnet_b0.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [1280, "efficientnet_b0", "DEFAULT", True, 2]] # (1, 1280, 20, 20) - 0
- [-1, 1, TorchVision, [1280, efficientnet_b0, DEFAULT, True, 2]] # (1, 1280, 20, 20) - 0

head:
- [-1, 1, Classify, [nc]]
2 changes: 1 addition & 1 deletion cfg/classify/mobilenet_v3_large.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [960, "mobilenet_v3_large", "DEFAULT", True, 2]] # (1, 960, 20, 20) - 0
- [-1, 1, TorchVision, [960, mobilenet_v3_large, DEFAULT, True, 2]] # (1, 960, 20, 20) - 0

head:
- [-1, 1, Classify, [nc]]
2 changes: 1 addition & 1 deletion cfg/classify/swin_t.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "swin_t", "DEFAULT", True, 3]] # (1, 768, 20, 20) - 0
- [-1, 1, TorchVision, [768, swin_t, DEFAULT, True, 3]] # (1, 768, 20, 20) - 0

head:
- [-1, 1, Classify, [nc]]
2 changes: 1 addition & 1 deletion cfg/detect/convnext_tiny.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [8, 16, 32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "convnext_tiny", "DEFAULT", True, 2, True]]
- [-1, 1, TorchVision, [768, convnext_tiny, DEFAULT, True, 2, True]]
- [0, 1, Index, [192, 4]] # (1, 192, 80, 80) - 1
- [0, 1, Index, [384, 6]] # (1, 384, 40, 40) - 2
- [0, 1, Index, [768, 8]] # (1, 768, 20, 20) - 3
Expand Down
37 changes: 37 additions & 0 deletions cfg/detect/mobilenet_v3_large-fpn.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
# Metadata
author: Y-T-G
task: detect
keywords: [cnn, mobilenet, fpn]
description: MobileNetv3-Large pretrained backbone with YOLO11 FPN neck and Detect head.
flops: 16.3
parameters: 14505363
min_version: 8.3.59
discussion: https://github.com/Y-T-G/community/discussions/51

# Model
nc: 1
strides: [8, 16, 32]
scales:
# [depth, width, max_channels]
default: [1.00, 1.00, 1024]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, mobilenet_v3_large, DEFAULT, True, 2, True]] # 0
- [0, 1, Index, [40, 7]] # (1, 40, 80, 80) - 1
- [0, 1, Index, [112, 13]] # (1, 112, 40, 40) - 2
- [0, 1, Index, [960, 17]] # (1, 960, 20, 20) - 3
- [-1, 1, SPPF, [960, 5]] # (1, 960, 20, 20) - 4
head:
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 960, 40, 40) - 5
- [[-1, 2], 1, Concat, [1]] # (1, 1072, 40, 40) - 6
- [-1, 2, C3k2, [112, False]] # (1, 112, 40, 40) - 7
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 112, 80, 80) - 8
- [[-1, 1], 1, Concat, [1]] # (1, 152, 80, 80) - 9
- [-1, 2, C3k2, [40, False]] # (1, 40, 80, 80) - 10
- [-1, 1, Conv, [40, 3, 2]] # (1, 40, 40, 40) - 11
- [[-1, -5], 1, Concat, [1]] # (1, 152, 40, 40) - 12
- [-1, 2, C3k2, [112, False]] # (1, 112, 40, 40) - 13
- [-1, 1, Conv, [112, 3, 2]] # (1, 112, 20, 20) - 14
- [[-1, -11], 1, Concat, [1]] # (1, 1072, 20, 20) - 15
- [-1, 2, C3k2, [960, True]] # (1, 960, 20, 20) - 16
- [[-7, -4, -1], 1, Detect, [nc]] # 17
37 changes: 37 additions & 0 deletions cfg/detect/mobilenet_v3_small-fpn.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
# Metadata
author: Y-T-G
task: detect
keywords: [cnn, mobilenet, fpn]
description: MobileNetv3-Small pretrained backbone with YOLO11 FPN neck and Detect head.
flops: 5.9
parameters: 5245459
min_version: 8.3.59
discussion: https://github.com/Y-T-G/community/discussions/53

# Model
nc: 1
strides: [8, 16, 32]
scales:
# [depth, width, max_channels]
default: [1.00, 1.00, 1024]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, mobilenet_v3_small, DEFAULT, True, 2, True]] # 0
- [0, 1, Index, [24, 4]] # (1, 40, 80, 80) - 1
- [0, 1, Index, [48, 9]] # (1, 112, 40, 40) - 2
- [0, 1, Index, [576, 13]] # (1, 960, 20, 20) - 3
- [-1, 1, SPPF, [578, 5]] # (1, 960, 20, 20) - 4
head:
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 960, 40, 40) - 5
- [[-1, 2], 1, Concat, [1]] # (1, 1072, 40, 40) - 6
- [-1, 2, C3k2, [48, False]] # (1, 112, 40, 40) - 7
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 112, 80, 80) - 8
- [[-1, 1], 1, Concat, [1]] # (1, 152, 80, 80) - 9
- [-1, 2, C3k2, [24, False]] # (1, 40, 80, 80) - 10
- [-1, 1, Conv, [24, 3, 2]] # (1, 40, 40, 40) - 11
- [[-1, -5], 1, Concat, [1]] # (1, 152, 40, 40) - 12
- [-1, 2, C3k2, [48, False]] # (1, 112, 40, 40) - 13
- [-1, 1, Conv, [48, 3, 2]] # (1, 112, 20, 20) - 14
- [[-1, -11], 1, Concat, [1]] # (1, 1072, 20, 20) - 15
- [-1, 2, C3k2, [576, True]] # (1, 960, 20, 20) - 16
- [[-7, -4, -1], 1, Detect, [nc]] # 17
2 changes: 1 addition & 1 deletion cfg/detect/resnet101.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [8, 16, 32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "resnet101", "DEFAULT", True, 2, True]]
- [-1, 1, TorchVision, [768, resnet101, DEFAULT, True, 2, True]]
- [0, 1, Index, [512, 6]] # (1, 512, 80, 80) - 1
- [0, 1, Index, [1024, 7]] # (1, 1024, 40, 40) - 2
- [0, 1, Index, [2048, 8]] # (1, 2048, 20, 20) - 3
Expand Down
40 changes: 40 additions & 0 deletions cfg/detect/swin_t-fpn.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
# Metadata
author: Y-T-G
task: detect
keywords: [transformer, swin, fpn]
description: Swin-T pretrained backbone with YOLO11 FPN neck and Detect head.
flops: 73.7
parameters: 40200397
min_version: 8.3.64
discussion: https://github.com/Y-T-G/community/discussions/41

# Model
nc: 1
strides: [8, 16, 32]
scales:
# [depth, width, max_channels]
default: [1.00, 1.00, 1024]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, swin_t, DEFAULT, True, 5, True]] # 0
- [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
- [-1, 1, SPPF, [768, 5]] # (1, 768, 20, 20) - 7
head:
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 768, 40, 40) - 8
- [[-1, 4], 1, Concat, [1]] # (1, 1152, 40, 40) - 9
- [-1, 2, C3k2, [384, False]] # (1, 384, 40, 40) - 10
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 384, 80, 80) - 11
- [[-1, 2], 1, Concat, [1]] # (1, 576, 80, 80) - 12
- [-1, 2, C3k2, [192, False]] # (1, 192, 80, 80) - 13
- [-1, 1, Conv, [192, 3, 2]] # (1, 192, 40, 40) - 14
- [[-1, -5], 1, Concat, [1]] # (1, 576, 40, 40) - 15
- [-1, 2, C3k2, [384, False]] # (1, 384, 40, 40) - 16
- [-1, 1, Conv, [384, 3, 2]] # (1, 384, 20, 20) - 17
- [[-1, -11], 1, Concat, [1]] # (1, 1152, 20, 20) - 18
- [-1, 2, C3k2, [768, True]] # (1, 768, 20, 20) - 19
- [[-7, -4, -1], 1, Detect, [nc]] # 20
2 changes: 1 addition & 1 deletion cfg/detect/swin_t.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [8, 16, 32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "swin_t", "DEFAULT", True, 5, True]]
- [-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
Expand Down
2 changes: 1 addition & 1 deletion cfg/obb/convnext_tiny.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [8, 16, 32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "convnext_tiny", "DEFAULT", True, 2, True]]
- [-1, 1, TorchVision, [768, convnext_tiny, DEFAULT, True, 2, True]]
- [0, 1, Index, [192, 4]] # (1, 192, 80, 80) - 1
- [0, 1, Index, [384, 6]] # (1, 384, 40, 40) - 2
- [0, 1, Index, [768, 8]] # (1, 768, 20, 20) - 3
Expand Down
39 changes: 39 additions & 0 deletions cfg/obb/swin_t-fpn.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
# Metadata
author: Y-T-G
task: obb
keywords: [transformer, swin, fpn]
description: Swin-T pretrained backbone with YOLO11 FPN neck and OBB head.
flops: 76.0
parameters: 40843936
min_version: 8.3.64

# Model
nc: 1
strides: [8, 16, 32]
scales:
# [depth, width, max_channels]
default: [1.00, 1.00, 1024]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, swin_t, DEFAULT, True, 5, True]] # 0
- [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
- [-1, 1, SPPF, [768, 5]] # (1, 768, 20, 20) - 7
head:
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 768, 40, 40) - 8
- [[-1, 4], 1, Concat, [1]] # (1, 1152, 40, 40) - 9
- [-1, 2, C3k2, [384, False]] # (1, 384, 40, 40) - 10
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 384, 80, 80) - 11
- [[-1, 2], 1, Concat, [1]] # (1, 576, 80, 80) - 12
- [-1, 2, C3k2, [192, False]] # (1, 192, 80, 80) - 13
- [-1, 1, Conv, [192, 3, 2]] # (1, 192, 40, 40) - 14
- [[-1, -5], 1, Concat, [1]] # (1, 576, 40, 40) - 15
- [-1, 2, C3k2, [384, False]] # (1, 384, 40, 40) - 16
- [-1, 1, Conv, [384, 3, 2]] # (1, 384, 20, 20) - 17
- [[-1, -11], 1, Concat, [1]] # (1, 1152, 20, 20) - 18
- [-1, 2, C3k2, [768, True]] # (1, 768, 20, 20) - 19
- [[-7, -4, -1], 1, OBB, [nc, 1]] # 20
2 changes: 1 addition & 1 deletion cfg/obb/swin_t.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [8, 16, 32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "swin_t", "DEFAULT", True, 5, True]]
- [-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
Expand Down
2 changes: 1 addition & 1 deletion cfg/pose/convnext_tiny.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ strides: [8, 16, 32]
kpt_shape: [17, 3]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "convnext_tiny", "DEFAULT", True, 2, True]]
- [-1, 1, TorchVision, [768, convnext_tiny, DEFAULT, True, 2, True]]
- [0, 1, Index, [192, 4]] # (1, 192, 80, 80) - 1
- [0, 1, Index, [384, 6]] # (1, 384, 40, 40) - 2
- [0, 1, Index, [768, 8]] # (1, 768, 20, 20) - 3
Expand Down
41 changes: 41 additions & 0 deletions cfg/pose/swin_t-fpn.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
# Metadata
author: Y-T-G
task: pose
keywords: [transformer, swin, fpn]
description: Swin-T pretrained backbone with YOLO11 FPN neck and Pose head.
flops: 76.2
parameters: 40896088
min_version: 8.3.64
discussion: https://github.com/Y-T-G/community/discussions/48

# Model
nc: 1
strides: [8, 16, 32]
kpt_shape: [17, 3]
scales:
# [depth, width, max_channels]
default: [1.00, 1.00, 1024]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, swin_t, DEFAULT, True, 5, True]] # 0
- [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
- [-1, 1, SPPF, [768, 5]] # (1, 768, 20, 20) - 7
head:
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 768, 40, 40) - 8
- [[-1, 4], 1, Concat, [1]] # (1, 1152, 40, 40) - 9
- [-1, 2, C3k2, [384, False]] # (1, 384, 40, 40) - 10
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 384, 80, 80) - 11
- [[-1, 2], 1, Concat, [1]] # (1, 576, 80, 80) - 12
- [-1, 2, C3k2, [192, False]] # (1, 192, 80, 80) - 13
- [-1, 1, Conv, [192, 3, 2]] # (1, 192, 40, 40) - 14
- [[-1, -5], 1, Concat, [1]] # (1, 576, 40, 40) - 15
- [-1, 2, C3k2, [384, False]] # (1, 384, 40, 40) - 16
- [-1, 1, Conv, [384, 3, 2]] # (1, 384, 20, 20) - 17
- [[-1, -11], 1, Concat, [1]] # (1, 1152, 20, 20) - 18
- [-1, 2, C3k2, [768, True]] # (1, 768, 20, 20) - 19
- [[-7, -4, -1], 1, Pose, [nc, kpt_shape]] # 20
2 changes: 1 addition & 1 deletion cfg/pose/swin_t.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ strides: [8, 16, 32]
kpt_shape: [17, 3]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "swin_t", "DEFAULT", True, 5, True]]
- [-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
Expand Down
2 changes: 1 addition & 1 deletion cfg/segment/convnext_tiny.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [8, 16, 32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "convnext_tiny", "DEFAULT", True, 2, True]]
- [-1, 1, TorchVision, [768, convnext_tiny, DEFAULT, True, 2, True]]
- [0, 1, Index, [192, 4]] # (1, 192, 80, 80) - 1
- [0, 1, Index, [384, 6]] # (1, 384, 40, 40) - 2
- [0, 1, Index, [768, 8]] # (1, 768, 20, 20) - 3
Expand Down
40 changes: 40 additions & 0 deletions cfg/segment/swin_t-fpn.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
# Metadata
author: Y-T-G
task: segment
keywords: [transformer, swin, fpn]
description: Swin-T pretrained backbone with YOLO11 FPN neck and Segment head.
flops: 125.8
parameters: 42152365
min_version: 8.3.64
discussion: https://github.com/Y-T-G/community/discussions/47

# Model
nc: 1
strides: [8, 16, 32]
scales:
# [depth, width, max_channels]
default: [1.00, 1.00, 1024]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, swin_t, DEFAULT, True, 5, True]] # 0
- [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
- [-1, 1, SPPF, [768, 5]] # (1, 768, 20, 20) - 7
head:
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 768, 40, 40) - 8
- [[-1, 4], 1, Concat, [1]] # (1, 1152, 40, 40) - 9
- [-1, 2, C3k2, [384, False]] # (1, 384, 40, 40) - 10
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 384, 80, 80) - 11
- [[-1, 2], 1, Concat, [1]] # (1, 576, 80, 80) - 12
- [-1, 2, C3k2, [192, False]] # (1, 192, 80, 80) - 13
- [-1, 1, Conv, [192, 3, 2]] # (1, 192, 40, 40) - 14
- [[-1, -5], 1, Concat, [1]] # (1, 576, 40, 40) - 15
- [-1, 2, C3k2, [384, False]] # (1, 384, 40, 40) - 16
- [-1, 1, Conv, [384, 3, 2]] # (1, 384, 20, 20) - 17
- [[-1, -11], 1, Concat, [1]] # (1, 1152, 20, 20) - 18
- [-1, 2, C3k2, [768, True]] # (1, 768, 20, 20) - 19
- [[-7, -4, -1], 1, Segment, [nc, 32, 256]] # 20
2 changes: 1 addition & 1 deletion cfg/segment/swin_t.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ nc: 1
strides: [8, 16, 32]
backbone:
# [from, repeats, module, args]
- [-1, 1, TorchVision, [768, "swin_t", "DEFAULT", True, 5, True]]
- [-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
Expand Down
Loading