Skip to content

Commit

Permalink
Add Swin-T FPN Pose config (#45)
Browse files Browse the repository at this point in the history
  • Loading branch information
Y-T-G authored Jan 21, 2025
1 parent 1b9bb3f commit a3065c3
Showing 1 changed file with 40 additions and 0 deletions.
40 changes: 40 additions & 0 deletions cfg/pose/swin_t-fpn.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
# 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

# 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

0 comments on commit a3065c3

Please sign in to comment.