-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathyolov5s-pruned-slim.yaml
20 lines (20 loc) · 1.61 KB
/
yolov5s-pruned-slim.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
{nc: 80, depth_multiple: 0.33, width_multiple: 0.5, anchors: [[10, 13, 16, 30, 33,
23], [30, 61, 62, 45, 59, 119], [116, 90, 156, 198, 373, 326]], backbone: [
[-1, 1, Conv, [64, 6, 2, 2, 1, true, 0.5]], [-1, 1, Conv, [128, 3, 2, None, 1,
true, 1.0]], [-1, 3, C3, [128, true, 1, [0.5, 0.375], [0.5, 1.0, 1.0], 0.625]],
[-1, 1, Conv, [256, 3, 2, None, 1, true, 0.75]], [-1, 6, C3, [256, true, 1, [
0.5, 0.25], [0.25, 0.375, 1.0, 1.0, 1.0, 1.0], 0.5]], [-1, 1, Conv, [512,
3, 2, None, 1, true, 0.25]], [-1, 9, C3, [512, true, 1, [0.5, 0.40625], [
0.125, 0.0625, 0.0625, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], 0.71875]], [-1, 1,
Conv, [1024, 3, 2, None, 1, true, 0.125]], [-1, 3, C3, [1024, true, 1, [0.5,
0.421875], [0.875, 1.0, 1.0], 0.859375]], [-1, 1, SPPF, [1024, 5, 0.03125]]],
head: [[-1, 1, Conv, [512, 1, 1, None, 1, true, 0.09375]], [-1, 1, nn.Upsample,
[None, 2, nearest]], [[-1, 6], 1, Concat, [1]], [-1, 3, C3, [512, false, 1,
[0.5, 0.34375], [0.0625, 1.0, 1.0], 0.78125]], [-1, 1, Conv, [256, 1, 1, None,
1, true, 0.125]], [-1, 1, nn.Upsample, [None, 2, nearest]], [[-1, 4], 1, Concat,
[1]], [-1, 3, C3, [256, false, 1, [0.5, 0.1875], [0.125, 1.0, 1.0], 0.875]],
[-1, 1, Conv, [256, 3, 2, None, 1, true, 0.5]], [[-1, 14], 1, Concat, [1]], [
-1, 3, C3, [512, false, 1, [0.5, 0.25], [0.125, 1.0, 1.0], 0.84375]], [-1, 1,
Conv, [512, 3, 2, None, 1, true, 0.21875]], [[-1, 10], 1, Concat, [1]], [-1,
3, C3, [1024, false, 1, [0.5, 0.328125], [0.21875, 1.0, 1.0], 0.890625]], [
[17, 20, 23], 1, Detect, [nc, anchors]]], ch: 3}