|
|
|
nc: 80 |
|
depth_multiple: 1.0 |
|
width_multiple: 1.0 |
|
|
|
|
|
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, Conv, [128, 3, 2]], |
|
[-1, 3, BottleneckCSPA, [128]], |
|
[-1, 1, Conv, [256, 3, 2]], |
|
[-1, 7, BottleneckCSPA, [256]], |
|
[-1, 1, Conv, [384, 3, 2]], |
|
[-1, 7, BottleneckCSPA, [384]], |
|
[-1, 1, Conv, [512, 3, 2]], |
|
[-1, 3, BottleneckCSPA, [512]], |
|
[-1, 1, Conv, [640, 3, 2]], |
|
[-1, 3, BottleneckCSPA, [640]], |
|
] |
|
|
|
|
|
head: |
|
[[-1, 1, SPPCSPC, [320]], |
|
[-1, 1, Conv, [256, 1, 1]], |
|
[-1, 1, nn.Upsample, [None, 2, 'nearest']], |
|
[-6, 1, Conv, [256, 1, 1]], |
|
[[-1, -2], 1, Concat, [1]], |
|
[-1, 3, BottleneckCSPB, [256]], |
|
[-1, 1, Conv, [192, 1, 1]], |
|
[-1, 1, nn.Upsample, [None, 2, 'nearest']], |
|
[-13, 1, Conv, [192, 1, 1]], |
|
[[-1, -2], 1, Concat, [1]], |
|
[-1, 3, BottleneckCSPB, [192]], |
|
[-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, Conv, [192, 3, 2]], |
|
[[-1, 22], 1, Concat, [1]], |
|
[-1, 3, BottleneckCSPB, [192]], |
|
[-1, 1, Conv, [384, 3, 1]], |
|
[-2, 1, Conv, [256, 3, 2]], |
|
[[-1, 17], 1, Concat, [1]], |
|
[-1, 3, BottleneckCSPB, [256]], |
|
[-1, 1, Conv, [512, 3, 1]], |
|
[-2, 1, Conv, [320, 3, 2]], |
|
[[-1, 12], 1, Concat, [1]], |
|
[-1, 3, BottleneckCSPB, [320]], |
|
[-1, 1, Conv, [640, 3, 1]], |
|
|
|
[[28,32,36,40], 1, IDetect, [nc, anchors]], |
|
] |