Skip to content

Commit 50e8d26

Browse files
authored
Add YOLO11-Pose P6 config (#68)
1 parent fcdb6a0 commit 50e8d26

File tree

1 file changed

+91
-0
lines changed

1 file changed

+91
-0
lines changed

cfg/pose/yolo11-p6.yaml

Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,91 @@
1+
# Metadata
2+
author: Y-T-G
3+
task: pose
4+
keywords: [cnn, yolo11, large-object]
5+
description: YOLO11 keypoint estimation model with an extra P6 scale for large object detecton.
6+
flops:
7+
n: 8.1
8+
s: 24.9
9+
m: 74.3
10+
l: 94.2
11+
x: 210.6
12+
parameters:
13+
n: 4790740
14+
s: 16822644
15+
m: 31257436
16+
l: 39076956
17+
x: 87749884
18+
min_version: 8.3.0
19+
20+
# Model
21+
nc: 80
22+
strides: [8, 16, 32, 64]
23+
kpt_shape: [17, 3]
24+
scales:
25+
n: [0.50, 0.25, 1024]
26+
s: [0.50, 0.50, 1024]
27+
m: [0.50, 1.00, 512]
28+
l: [1.00, 1.00, 512]
29+
x: [1.00, 1.50, 512]
30+
31+
backbone:
32+
# Stem
33+
- [-1, 1, Conv, [64, 3, 2]] # (1, 64, 320, 320) - 0
34+
35+
# P2 Backbone
36+
- [-1, 1, Conv, [128, 3, 2]] # (1, 128, 160, 160) - 1
37+
- [-1, 2, C3k2, [256, False, 0.25]] # (1, 256, 160, 160) - 2
38+
39+
# P3 Backbone
40+
- [-1, 1, Conv, [256, 3, 2]] # (1, 256, 80, 80) - 3
41+
- [-1, 2, C3k2, [512, False, 0.25]] # (1, 512, 80, 80) - 4
42+
43+
# P4 Backbone
44+
- [-1, 1, Conv, [512, 3, 2]] # (1, 512, 40, 40) - 5
45+
- [-1, 2, C3k2, [768, True]] # (1, 512, 40, 40) - 6
46+
47+
# P5 Backbone
48+
- [-1, 1, Conv, [768, 3, 2]] # (1, 512, 20, 20) - 7
49+
- [-1, 2, C3k2, [768, True]] # (1, 512, 20, 20) - 8
50+
51+
# P6 Backbone
52+
- [-1, 1, Conv, [1024, 3, 2]] # (1, 512, 10, 10) - 9
53+
- [-1, 2, C3k2, [1024, True]] # (1, 512, 10, 10) - 10
54+
- [-1, 1, SPPF, [1024, 5]] # (1, 512, 10, 10) - 11
55+
- [-1, 2, C2PSA, [1024]] # (1, 512, 10, 10) - 12
56+
57+
head:
58+
# Top-Down Path (Feature Pyramid)
59+
# P5 Head
60+
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 512, 20, 20) - 13
61+
- [[-1, 8], 1, Concat, [1]] # (1, 1024, 20, 20) - 14
62+
- [-1, 2, C3k2, [768, False]] # (1, 512, 20, 20) - 15
63+
64+
# P4 Head
65+
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 512, 40, 40) - 16
66+
- [[-1, 6], 1, Concat, [1]] # (1, 1024, 40, 40) - 17
67+
- [-1, 2, C3k2, [512, False]] # (1, 512, 40, 40) - 18
68+
69+
# P3 Head
70+
- [-1, 1, nn.Upsample, [None, 2, nearest]] # (1, 512, 80, 80) - 19
71+
- [[-1, 4], 1, Concat, [1]] # (1, 1024, 80, 80) - 20
72+
- [-1, 2, C3k2, [256, False]] # (1, 256, 80, 80) - 21
73+
74+
# Bottom-Up Path (PAN-FPN)
75+
# P4
76+
- [-1, 1, Conv, [256, 3, 2]] # (1, 256, 40, 40) - 22
77+
- [[-1, 16], 1, Concat, [1]] # (1, 768, 40, 40) - 23
78+
- [-1, 2, C3k2, [512, False]] # (1, 512, 40, 40) - 24
79+
80+
# P5
81+
- [-1, 1, Conv, [512, 3, 2]] # (1, 512, 20, 20) - 25
82+
- [[-1, 13], 1, Concat, [1]] # (1, 1024, 20, 20) - 26
83+
- [-1, 2, C3k2, [768, True]] # (1, 512, 20, 20) - 27
84+
85+
# P6
86+
- [-1, 1, Conv, [768, 3, 2]] # (1, 512, 10, 10) - 28
87+
- [[-1, 10], 1, Concat, [1]] # (1, 1024, 10, 10) - 29
88+
- [-1, 2, C3k2, [1024, True]] # (1, 512, 10, 10) - 30
89+
90+
# Pose Head (P3, P4, P5, P6)
91+
- [[21, 24, 27, 30], 1, Pose, [nc, kpt_shape]] # 31

0 commit comments

Comments
 (0)