|
|
|
@ -30,17 +30,17 @@ backbone: |
|
|
|
|
# YOLOv5 v6.0 head |
|
|
|
|
head: |
|
|
|
|
- [-1, 1, Conv, [768, 1, 1]] |
|
|
|
|
- [-1, 1, nn.Upsample, [None, 2, 'nearest']] |
|
|
|
|
- [-1, 1, nn.Upsample, [None, 2, "nearest"]] |
|
|
|
|
- [[-1, 8], 1, Concat, [1]] # cat backbone P5 |
|
|
|
|
- [-1, 3, C3, [768, False]] # 15 |
|
|
|
|
|
|
|
|
|
- [-1, 1, Conv, [512, 1, 1]] |
|
|
|
|
- [-1, 1, nn.Upsample, [None, 2, 'nearest']] |
|
|
|
|
- [-1, 1, nn.Upsample, [None, 2, "nearest"]] |
|
|
|
|
- [[-1, 6], 1, Concat, [1]] # cat backbone P4 |
|
|
|
|
- [-1, 3, C3, [512, False]] # 19 |
|
|
|
|
|
|
|
|
|
- [-1, 1, Conv, [256, 1, 1]] |
|
|
|
|
- [-1, 1, nn.Upsample, [None, 2, 'nearest']] |
|
|
|
|
- [-1, 1, nn.Upsample, [None, 2, "nearest"]] |
|
|
|
|
- [[-1, 4], 1, Concat, [1]] # cat backbone P3 |
|
|
|
|
- [-1, 3, C3, [256, False]] # 23 (P3/8-small) |
|
|
|
|
|
|
|
|
|