#-1 means layer_id of input_data
#1 means repeat times of this block
yolov4_blocks_args = [
[-1, 1, “Conv”, [32, 3, 1]], # 0
[-1, 1, “Conv”, [64, 3, 2]], # 1-P1/2
[-1, 1, “Bottleneck”, [64]],
[-1, 1, “Conv”, [128, 3, 2]], # 3-P2/4
[-1, 2, “BottleneckCSP”, [128]],
[-1, 1, “Conv”, [256, 3, 2]], # 5-P3/8
[-1, 8, “BottleneckCSP”, [256]],
[-1, 1, “Conv”, [512, 3, 2]], # 7-P4/16
[-1, 8, “BottleneckCSP”, [512]],
[-1, 1, “Conv”, [1024, 3, 2]], # 9-P5/32
[-1, 4, “BottleneckCSP”, [1024]], # 10
# head
[-1, 1, “SPPCSP”, [512]], # 11
[-1, 1, “Conv”, [256, 1, 1]],
[-1, 1, “Upsample”, [None, 2, ‘nearest’]], # 13
[8, 1, “Conv”, [256, 1, 1]], # route backbone P4 # 14
[[-1, 13], 1, “Concat”, [1]],
[-1, 2, “BottleneckCSP2”, [256]], # 16
[-1, 1, “Conv”, [128, 1, 1]],
[-1, 1, “Upsample”, [None, 2, ‘nearest’]],
[6, 1, “Conv”, [128, 1, 1]], # route backbone P3 # 19
[[-1, 18], 1, “Concat”, [1]],
[-1, 2, “BottleneckCSP2”, [128]], # 21
[-1, 1, “Conv”, [256, 3, 1]], # (P3/8-small) 22
[21, 1, “Conv”, [256, 3, 2]],
[[-1, 16], 1, “Concat”, [1]], # cat
[-1, 2, “BottleneckCSP2”, [256]], # 25
[-1, 1, “Conv”, [512, 3, 1]], #  (P4/16-medium) 26
[25, 1, “Conv”, [512, 3, 2]],
[[-1, 11], 1, “Concat”, [1]], # cat
[-1, 2, “BottleneckCSP2”, [512]], # 29
[-1, 1, “Conv”, [1024, 3, 1]], # (P5/32-large)
# 22,26,30
]
2021-07-28
最新推荐文章于 2022-06-01 12:17:40 发布