一、backbone与head
yolov5中backbone:
# [from, number, module, args]
[[-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2
[-1, 1, Conv, [128, 3, 2]], # 1-P2/4
[-1, 3, C3, [128]],
[-1, 1, Conv, [256, 3, 2]], # 3-P3/8
[-1, 6, C3, [256]],
[-1, 1, Conv, [512, 3, 2]], # 5-P4/16
[-1, 9, C3, [512]],
[-1, 1, Conv, [1024, 3, 2]], # 7-P5/32
[-1, 3, C3, [1024]],
[-1, 1, SPPF, [1024, 5]], # 9
]
from:表示与上层连接
number:模块重复次数
modul:模块类型
args:卷积层参数信息
其中,存在“#0-P1/2,# 1-P2/4..”标记,
"0","1"..表示当成层的序号,在head模块中用到并且+1
“P1/2,P2/4”表示经过该层,feature map的尺寸缩减为原图2倍、4倍。。
其中C3模块结构形式: