Hugging Face's logo

Spaces:
xiang-wuu
/
yolov5
Runtime error

App Files Community
glenn-jocher commited on
Commit
4ba47b1
·
1 Parent(s): 1e2cb6b

model.yaml updates

Browse files
Files changed (5) hide show
  1. models/yolov3-spp.yaml +2 -2
  2. models/yolov5l.yaml +3 -3
  3. models/yolov5m.yaml +3 -3
  4. models/yolov5s.yaml +3 -3
  5. models/yolov5x.yaml +3 -3
models/yolov3-spp.yaml CHANGED
@@ -1,7 +1,7 @@
1
  # parameters
2
  nc: 80 # number of classes
3
- depth_multiple: 1.0 # expand model depth
4
- width_multiple: 1.0 # expand layer channels
5
 
6
  # anchors
7
  anchors:
 
1
  # parameters
2
  nc: 80 # number of classes
3
+ depth_multiple: 1.0 # model depth multiple
4
+ width_multiple: 1.0 # layer channel multiple
5
 
6
  # anchors
7
  anchors:
models/yolov5l.yaml CHANGED
@@ -27,19 +27,19 @@ backbone:
27
  # yolov5 head
28
  head:
29
  [[-1, 3, BottleneckCSP, [1024, False]], # 11
30
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 12 (P5/32-large)
31
 
32
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
33
  [[-1, 6], 1, Concat, [1]], # cat backbone P4
34
  [-1, 1, Conv, [512, 1, 1]],
35
  [-1, 3, BottleneckCSP, [512, False]],
36
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 17 (P4/16-medium)
37
 
38
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
39
  [[-1, 4], 1, Concat, [1]], # cat backbone P3
40
  [-1, 1, Conv, [256, 1, 1]],
41
  [-1, 3, BottleneckCSP, [256, False]],
42
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 22 (P3/8-small)
43
 
44
  [[], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
45
  ]
 
27
  # yolov5 head
28
  head:
29
  [[-1, 3, BottleneckCSP, [1024, False]], # 11
30
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 12 (P5/32-large)
31
 
32
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
33
  [[-1, 6], 1, Concat, [1]], # cat backbone P4
34
  [-1, 1, Conv, [512, 1, 1]],
35
  [-1, 3, BottleneckCSP, [512, False]],
36
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 17 (P4/16-medium)
37
 
38
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
39
  [[-1, 4], 1, Concat, [1]], # cat backbone P3
40
  [-1, 1, Conv, [256, 1, 1]],
41
  [-1, 3, BottleneckCSP, [256, False]],
42
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 22 (P3/8-small)
43
 
44
  [[], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
45
  ]
models/yolov5m.yaml CHANGED
@@ -27,19 +27,19 @@ backbone:
27
  # yolov5 head
28
  head:
29
  [[-1, 3, BottleneckCSP, [1024, False]], # 11
30
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 12 (P5/32-large)
31
 
32
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
33
  [[-1, 6], 1, Concat, [1]], # cat backbone P4
34
  [-1, 1, Conv, [512, 1, 1]],
35
  [-1, 3, BottleneckCSP, [512, False]],
36
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 17 (P4/16-medium)
37
 
38
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
39
  [[-1, 4], 1, Concat, [1]], # cat backbone P3
40
  [-1, 1, Conv, [256, 1, 1]],
41
  [-1, 3, BottleneckCSP, [256, False]],
42
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 22 (P3/8-small)
43
 
44
  [[], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
45
  ]
 
27
  # yolov5 head
28
  head:
29
  [[-1, 3, BottleneckCSP, [1024, False]], # 11
30
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 12 (P5/32-large)
31
 
32
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
33
  [[-1, 6], 1, Concat, [1]], # cat backbone P4
34
  [-1, 1, Conv, [512, 1, 1]],
35
  [-1, 3, BottleneckCSP, [512, False]],
36
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 17 (P4/16-medium)
37
 
38
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
39
  [[-1, 4], 1, Concat, [1]], # cat backbone P3
40
  [-1, 1, Conv, [256, 1, 1]],
41
  [-1, 3, BottleneckCSP, [256, False]],
42
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 22 (P3/8-small)
43
 
44
  [[], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
45
  ]
models/yolov5s.yaml CHANGED
@@ -27,19 +27,19 @@ backbone:
27
  # yolov5 head
28
  head:
29
  [[-1, 3, BottleneckCSP, [1024, False]], # 11
30
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 12 (P5/32-large)
31
 
32
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
33
  [[-1, 6], 1, Concat, [1]], # cat backbone P4
34
  [-1, 1, Conv, [512, 1, 1]],
35
  [-1, 3, BottleneckCSP, [512, False]],
36
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 17 (P4/16-medium)
37
 
38
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
39
  [[-1, 4], 1, Concat, [1]], # cat backbone P3
40
  [-1, 1, Conv, [256, 1, 1]],
41
  [-1, 3, BottleneckCSP, [256, False]],
42
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 22 (P3/8-small)
43
 
44
  [[], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
45
  ]
 
27
  # yolov5 head
28
  head:
29
  [[-1, 3, BottleneckCSP, [1024, False]], # 11
30
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 12 (P5/32-large)
31
 
32
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
33
  [[-1, 6], 1, Concat, [1]], # cat backbone P4
34
  [-1, 1, Conv, [512, 1, 1]],
35
  [-1, 3, BottleneckCSP, [512, False]],
36
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 17 (P4/16-medium)
37
 
38
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
39
  [[-1, 4], 1, Concat, [1]], # cat backbone P3
40
  [-1, 1, Conv, [256, 1, 1]],
41
  [-1, 3, BottleneckCSP, [256, False]],
42
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 22 (P3/8-small)
43
 
44
  [[], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
45
  ]
models/yolov5x.yaml CHANGED
@@ -27,19 +27,19 @@ backbone:
27
  # yolov5 head
28
  head:
29
  [[-1, 3, BottleneckCSP, [1024, False]], # 11
30
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 12 (P5/32-large)
31
 
32
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
33
  [[-1, 6], 1, Concat, [1]], # cat backbone P4
34
  [-1, 1, Conv, [512, 1, 1]],
35
  [-1, 3, BottleneckCSP, [512, False]],
36
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 17 (P4/16-medium)
37
 
38
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
39
  [[-1, 4], 1, Concat, [1]], # cat backbone P3
40
  [-1, 1, Conv, [256, 1, 1]],
41
  [-1, 3, BottleneckCSP, [256, False]],
42
- [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 22 (P3/8-small)
43
 
44
  [[], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
45
  ]
 
27
  # yolov5 head
28
  head:
29
  [[-1, 3, BottleneckCSP, [1024, False]], # 11
30
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 12 (P5/32-large)
31
 
32
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
33
  [[-1, 6], 1, Concat, [1]], # cat backbone P4
34
  [-1, 1, Conv, [512, 1, 1]],
35
  [-1, 3, BottleneckCSP, [512, False]],
36
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 17 (P4/16-medium)
37
 
38
  [-2, 1, nn.Upsample, [None, 2, 'nearest']],
39
  [[-1, 4], 1, Concat, [1]], # cat backbone P3
40
  [-1, 1, Conv, [256, 1, 1]],
41
  [-1, 3, BottleneckCSP, [256, False]],
42
+ [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1]], # 22 (P3/8-small)
43
 
44
  [[], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
45
  ]