-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathyolov5s_DLA_rep_ECA_weights.yaml
62 lines (47 loc) · 2.53 KB
/
yolov5s_DLA_rep_ECA_weights.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
# Parameters
nc: 80 # number of classes
depth_multiple: 0.33 # model depth multiple
width_multiple: 0.50 # layer channel multiple
anchors:
- [ 10,13, 16,30, 33,23 ] # P3/8
- [ 30,61, 62,45, 59,119 ] # P4/16
- [ 116,90, 156,198, 373,326 ] # P5/32
# YOLOv5 v6.0 backbone
backbone:
# [from, number, module, args] # 3, 256, 256
[ [ -1, 1, Conv, [ 64, 3, 2 ] ], # 0-P1/2 32, 128, 128
[ -1, 1, Tree, [ 1, BasicBlock_Rep3, 32, 64, 2, True ] ], # 1-P2/4 64, 64, 64
[ -1, 1, Tree, [ 2, BasicBlock_Rep3, 64, 128, 2, True ] ], # 2-P3/8 128, 32, 32
[ -1, 1, Tree, [ 2, BasicBlock_Rep3, 128, 256, 2, True ] ], # 3-P4/16 256, 16, 16
[ [ 2, 3 ], 1, DLAUp, [ [ 128, 256 ], [ 1, 2 ] ] ], # 4 128, 32, 32
[ -2, 1, Tree, [ 1, BasicBlock_Rep3, 256, 512, 2, True ] ], # 5-P5/32 512, 8, 8
[ [ 3, 5 ], 1, DLAUp, [ [ 256, 512 ], [ 1, 2 ] ] ], # 6 256, 16, 16
[ [ 4, 6 ], 1, DLAUp, [ [ 128, 256 ], [ 1, 2 ] ] ], # 7 128, 32, 32
[ -3, 1, SPPF, [ 1024, 5 ] ], # 8 512, 8, 8
[ [ 5, -1 ], 1, DLAUp, [ [ 512, 512 ], [ 1, 1 ] ] ], # 9 512, 8, 8
[ [ 6, -1 ], 1, DLAUp, [ [ 256, 512 ], [ 1, 2 ] ] ], # 10 256, 16, 16
[ [ 7, -1 ], 1, DLAUp, [ [ 128, 256 ], [ 1, 2 ] ] ], # 11 128, 32, 32
]
# YOLOv5 v6.0 head
head:
[ [ 9, 1, Conv, [ 512, 1, 1 ] ], # 256, 8, 8
[ -1, 1, nn.Upsample, [ None, 2, 'nearest' ] ], # 256, 16, 16
[ [ -1, 10 ], 1, Concat, [ 1 ] ], # cat backbone P4 512, 16, 16
[ -1, 3, C3, [ 512, False ] ], # 15 256, 16, 16
[ -1, 1, Conv, [ 256, 1, 1 ] ], # 128, 16, 16
[ -1, 1, nn.Upsample, [ None, 2, 'nearest' ] ], # 128, 32, 32
[ [ -1, 11 ], 1, Concat, [ 1 ] ], # 18 cat backbone P3 256, 32, 32
[ [ 16, 18 ], 1, Fusion_ECA, [ 256, 128, 'downsample' ] ], # 19 128, 16, 16
[ [ 12, 19 ], 1, Fusion_ECA, [ 128, 256, 'downsample' ] ], # 20 256, 8, 8
[ [ 19, 20 ], 1, Fusion_ECA, [ 256, 128, 'upsample' ] ], # 21 128, 16, 16
[ [ 18, 21 ], 1, Fusion_ECA, [ 128, 256, 'upsample' ] ], # 22 256, 32, 32
[ -1, 3, C3, [ 256, False ] ], # 23 (P3/8-small) 128, 32, 32
[ -1, 1, Conv, [ 256, 3, 2 ] ], # 128, 16, 16
[ [ -1, 21 ], 1, Concat, [ 1 ] ], # cat head P4 256, 16, 16
[ -1, 3, C3, [ 512, False ] ], # 26 (P4/16-medium) 256, 16, 16
[ -1, 1, Conv, [ 512, 3, 2 ] ], # 256, 8, 8
[ [ -1, 20 ], 1, Concat, [ 1 ] ], # cat head P5 512, 8, 8
[ -1, 3, C3, [ 1024, False ] ], # 29 (P5/32-large) 512, 8, 8
[ [ 23, 26, 29 ], 1, Detect, [ nc, anchors ] ], # Detect(P3, P4, P5)
]