123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263 |
- nc: 80
- depth_multiple: 1.0
- width_multiple: 1.25
- anchors:
- - [ 19,27, 44,40, 38,94 ]
- - [ 96,68, 86,152, 180,137 ]
- - [ 140,301, 303,264, 238,542 ]
- - [ 436,615, 739,380, 925,792 ]
- backbone:
-
- [[-1, 1, ReOrg, []],
- [-1, 1, Conv, [64, 3, 1]],
- [-1, 1, DownC, [128]],
- [-1, 3, BottleneckCSPA, [128]],
- [-1, 1, DownC, [256]],
- [-1, 15, BottleneckCSPA, [256]],
- [-1, 1, DownC, [512]],
- [-1, 15, BottleneckCSPA, [512]],
- [-1, 1, DownC, [768]],
- [-1, 7, BottleneckCSPA, [768]],
- [-1, 1, DownC, [1024]],
- [-1, 7, BottleneckCSPA, [1024]],
- ]
- head:
- [[-1, 1, SPPCSPC, [512]],
- [-1, 1, Conv, [384, 1, 1]],
- [-1, 1, nn.Upsample, [None, 2, 'nearest']],
- [-6, 1, Conv, [384, 1, 1]],
- [[-1, -2], 1, Concat, [1]],
- [-1, 3, BottleneckCSPB, [384]],
- [-1, 1, Conv, [256, 1, 1]],
- [-1, 1, nn.Upsample, [None, 2, 'nearest']],
- [-13, 1, Conv, [256, 1, 1]],
- [[-1, -2], 1, Concat, [1]],
- [-1, 3, BottleneckCSPB, [256]],
- [-1, 1, Conv, [128, 1, 1]],
- [-1, 1, nn.Upsample, [None, 2, 'nearest']],
- [-20, 1, Conv, [128, 1, 1]],
- [[-1, -2], 1, Concat, [1]],
- [-1, 3, BottleneckCSPB, [128]],
- [-1, 1, Conv, [256, 3, 1]],
- [-2, 1, DownC, [256]],
- [[-1, 22], 1, Concat, [1]],
- [-1, 3, BottleneckCSPB, [256]],
- [-1, 1, Conv, [512, 3, 1]],
- [-2, 1, DownC, [384]],
- [[-1, 17], 1, Concat, [1]],
- [-1, 3, BottleneckCSPB, [384]],
- [-1, 1, Conv, [768, 3, 1]],
- [-2, 1, DownC, [512]],
- [[-1, 12], 1, Concat, [1]],
- [-1, 3, BottleneckCSPB, [512]],
- [-1, 1, Conv, [1024, 3, 1]],
- [[28,32,36,40], 1, IDetect, [nc, anchors]],
- ]
|