|
nc: 3 # number of classes scale ‘n’ # [depth, width, max_channels] n: [0.33, 0.25, 1024] |
| backbone: # [from, repeats, module, args] - [−1, 1, Conv, [64, 3, 2]] # 0-P1/2 - [−1, 1, SPDConv, [128]] # 1-P2/4 - [−1, 3, C2f, [128, True]] - [−1, 1, SPDConv, [256]] # 3-P3/8 - [−1, 6, C2f, [256, True]] - [−1, 1, SPDConv, [512]] # 5-P4/16 - [−1, 6, C2f, [512, True]] - [−1, 1, SPDConv, [1024]] # 7-P5/32 - [−1, 3, C2f, [1024, True]] - [−1, 1, SegNext_Attention, []] # 9 - [−1, 1, SPPF, [1024, 5]] # 10 |
|
# YOLOv8.0-p2 head head: - [−1, 1, nn.Upsample, [None, 2, ‘nearest’]] - [[−1, 6], 1, Concat, [1]] # cat backbone P4 - [−1, 3, C2f, [512]] # 13 |
| - [−1, 1, nn.Upsample, [None, 2, ‘nearest’]] - [[−1, 4], 1, Concat, [1]] # cat backbone P3 - [−1, 3, C2f, [256]] # 16 (P3/8-small) |
| - [−1, 1, nn.Upsample, [None, 2, ‘nearest’]] - [[−1, 2], 1, Concat, [1]] # cat backbone P2 - [−1, 1, EMA, []] - [−1, 3, C2f, [128]] # 20 (P2/4-xsmall) 20 |
|
- [−1, 1, SPDConv, [128]] - [[−1, 16], 1, Concat, [1]] # cat head P3 - [−1, 3, C2f, [256]] # 23 (P3/8-small) 23 |
|
- [−1, 1, SPDConv, [256]] - [[−1, 13], 1, Concat, [1]] # cat head P4 - [−1, 3, C2f, [512]] # 26 (P4/16-medium) 26 |
| - [−1, 1, SPDConv, [512]] - [[−1, 10], 1, Concat, [1]] # cat head P5 - [−1, 3, C2f, [1024]] # 29 (P5/32-large) 29 |
| - [[20, 23, 26, 29], 1, Detect_DyHead, [nc, 128, 1]] # Detect(P2, P3, P4, P5) |