Commit
·
2268f9c
1
Parent(s):
adaf1d8
update *.yaml output layer comments
Browse files- models/yolov5l.yaml +3 -3
- models/yolov5m.yaml +3 -3
- models/yolov5s.yaml +3 -3
- models/yolov5x.yaml +3 -3
models/yolov5l.yaml
CHANGED
|
@@ -34,15 +34,15 @@ head:
|
|
| 34 |
[-1, 1, Conv, [256, 1, 1]],
|
| 35 |
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
|
| 36 |
[[-1, 4], 1, Concat, [1]], # cat backbone P3
|
| 37 |
-
[-1, 3, BottleneckCSP, [256, False]], #
|
| 38 |
|
| 39 |
[-1, 1, Conv, [256, 3, 2]],
|
| 40 |
[[-1, 14], 1, Concat, [1]], # cat head P4
|
| 41 |
-
[-1, 3, BottleneckCSP, [512, False]], # 20
|
| 42 |
|
| 43 |
[-1, 1, Conv, [512, 3, 2]],
|
| 44 |
[[-1, 10], 1, Concat, [1]], # cat head P5
|
| 45 |
-
[-1, 3, BottleneckCSP, [1024, False]], # 23
|
| 46 |
|
| 47 |
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
|
| 48 |
]
|
|
|
|
| 34 |
[-1, 1, Conv, [256, 1, 1]],
|
| 35 |
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
|
| 36 |
[[-1, 4], 1, Concat, [1]], # cat backbone P3
|
| 37 |
+
[-1, 3, BottleneckCSP, [256, False]], # 17 (P3/8-small)
|
| 38 |
|
| 39 |
[-1, 1, Conv, [256, 3, 2]],
|
| 40 |
[[-1, 14], 1, Concat, [1]], # cat head P4
|
| 41 |
+
[-1, 3, BottleneckCSP, [512, False]], # 20 (P4/16-medium)
|
| 42 |
|
| 43 |
[-1, 1, Conv, [512, 3, 2]],
|
| 44 |
[[-1, 10], 1, Concat, [1]], # cat head P5
|
| 45 |
+
[-1, 3, BottleneckCSP, [1024, False]], # 23 (P5/32-large)
|
| 46 |
|
| 47 |
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
|
| 48 |
]
|
models/yolov5m.yaml
CHANGED
|
@@ -34,15 +34,15 @@ head:
|
|
| 34 |
[-1, 1, Conv, [256, 1, 1]],
|
| 35 |
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
|
| 36 |
[[-1, 4], 1, Concat, [1]], # cat backbone P3
|
| 37 |
-
[-1, 3, BottleneckCSP, [256, False]], #
|
| 38 |
|
| 39 |
[-1, 1, Conv, [256, 3, 2]],
|
| 40 |
[[-1, 14], 1, Concat, [1]], # cat head P4
|
| 41 |
-
[-1, 3, BottleneckCSP, [512, False]], # 20
|
| 42 |
|
| 43 |
[-1, 1, Conv, [512, 3, 2]],
|
| 44 |
[[-1, 10], 1, Concat, [1]], # cat head P5
|
| 45 |
-
[-1, 3, BottleneckCSP, [1024, False]], # 23
|
| 46 |
|
| 47 |
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
|
| 48 |
]
|
|
|
|
| 34 |
[-1, 1, Conv, [256, 1, 1]],
|
| 35 |
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
|
| 36 |
[[-1, 4], 1, Concat, [1]], # cat backbone P3
|
| 37 |
+
[-1, 3, BottleneckCSP, [256, False]], # 17 (P3/8-small)
|
| 38 |
|
| 39 |
[-1, 1, Conv, [256, 3, 2]],
|
| 40 |
[[-1, 14], 1, Concat, [1]], # cat head P4
|
| 41 |
+
[-1, 3, BottleneckCSP, [512, False]], # 20 (P4/16-medium)
|
| 42 |
|
| 43 |
[-1, 1, Conv, [512, 3, 2]],
|
| 44 |
[[-1, 10], 1, Concat, [1]], # cat head P5
|
| 45 |
+
[-1, 3, BottleneckCSP, [1024, False]], # 23 (P5/32-large)
|
| 46 |
|
| 47 |
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
|
| 48 |
]
|
models/yolov5s.yaml
CHANGED
|
@@ -34,15 +34,15 @@ head:
|
|
| 34 |
[-1, 1, Conv, [256, 1, 1]],
|
| 35 |
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
|
| 36 |
[[-1, 4], 1, Concat, [1]], # cat backbone P3
|
| 37 |
-
[-1, 3, BottleneckCSP, [256, False]], #
|
| 38 |
|
| 39 |
[-1, 1, Conv, [256, 3, 2]],
|
| 40 |
[[-1, 14], 1, Concat, [1]], # cat head P4
|
| 41 |
-
[-1, 3, BottleneckCSP, [512, False]], # 20
|
| 42 |
|
| 43 |
[-1, 1, Conv, [512, 3, 2]],
|
| 44 |
[[-1, 10], 1, Concat, [1]], # cat head P5
|
| 45 |
-
[-1, 3, BottleneckCSP, [1024, False]], # 23
|
| 46 |
|
| 47 |
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
|
| 48 |
]
|
|
|
|
| 34 |
[-1, 1, Conv, [256, 1, 1]],
|
| 35 |
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
|
| 36 |
[[-1, 4], 1, Concat, [1]], # cat backbone P3
|
| 37 |
+
[-1, 3, BottleneckCSP, [256, False]], # 17 (P3/8-small)
|
| 38 |
|
| 39 |
[-1, 1, Conv, [256, 3, 2]],
|
| 40 |
[[-1, 14], 1, Concat, [1]], # cat head P4
|
| 41 |
+
[-1, 3, BottleneckCSP, [512, False]], # 20 (P4/16-medium)
|
| 42 |
|
| 43 |
[-1, 1, Conv, [512, 3, 2]],
|
| 44 |
[[-1, 10], 1, Concat, [1]], # cat head P5
|
| 45 |
+
[-1, 3, BottleneckCSP, [1024, False]], # 23 (P5/32-large)
|
| 46 |
|
| 47 |
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
|
| 48 |
]
|
models/yolov5x.yaml
CHANGED
|
@@ -34,15 +34,15 @@ head:
|
|
| 34 |
[-1, 1, Conv, [256, 1, 1]],
|
| 35 |
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
|
| 36 |
[[-1, 4], 1, Concat, [1]], # cat backbone P3
|
| 37 |
-
[-1, 3, BottleneckCSP, [256, False]], #
|
| 38 |
|
| 39 |
[-1, 1, Conv, [256, 3, 2]],
|
| 40 |
[[-1, 14], 1, Concat, [1]], # cat head P4
|
| 41 |
-
[-1, 3, BottleneckCSP, [512, False]], # 20
|
| 42 |
|
| 43 |
[-1, 1, Conv, [512, 3, 2]],
|
| 44 |
[[-1, 10], 1, Concat, [1]], # cat head P5
|
| 45 |
-
[-1, 3, BottleneckCSP, [1024, False]], # 23
|
| 46 |
|
| 47 |
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
|
| 48 |
]
|
|
|
|
| 34 |
[-1, 1, Conv, [256, 1, 1]],
|
| 35 |
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
|
| 36 |
[[-1, 4], 1, Concat, [1]], # cat backbone P3
|
| 37 |
+
[-1, 3, BottleneckCSP, [256, False]], # 17 (P3/8-small)
|
| 38 |
|
| 39 |
[-1, 1, Conv, [256, 3, 2]],
|
| 40 |
[[-1, 14], 1, Concat, [1]], # cat head P4
|
| 41 |
+
[-1, 3, BottleneckCSP, [512, False]], # 20 (P4/16-medium)
|
| 42 |
|
| 43 |
[-1, 1, Conv, [512, 3, 2]],
|
| 44 |
[[-1, 10], 1, Concat, [1]], # cat head P5
|
| 45 |
+
[-1, 3, BottleneckCSP, [1024, False]], # 23 (P5/32-large)
|
| 46 |
|
| 47 |
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
|
| 48 |
]
|