Skip to content

Commit 4b96423

Browse files
authored
Add YOLO11-OBB P6 config (#67)
1 parent 50e8d26 commit 4b96423

File tree

1 file changed

+90
-0
lines changed

1 file changed

+90
-0
lines changed

cfg/obb/yolo11-p6.yaml

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

0 commit comments

Comments
 (0)