针对yolov5的yaml文件,在改动模型时,其中的参数容易出错。
在此处可打印出一些信息,
例如,yolov5s模型, depth_multiple: 0.33 ,width_multiple: 0.50
[[-1, 1, Conv, [64, 6, 2, 2]],
-1 表示连接至上一层(若是 4 则表示连接至第4层)
1×0.33=0.33<1 则表示该模块的数量为 1(n = n_ = max(round(n * gd), 1) if n > 1 else n # depth gain,在yolo.py 中有这行代码来计算,计算出的值小于 1则按1算,若大于1则四舍五入取整)
输入通道数由上一层输出通道数决定
64×width_multiple=32 为 输出通道数
6 为卷积核大小
2 为步长
2 为填充值
[-1, 1, Conv, [128, 3, 2]],
即输出通道为 64 ,卷积核大小为 3, 步长为 2,填充为默认值
[-1, 3, C3, [128]],
输出通道数为 64
具体这些参数之间应该怎么联系,还在理解。。。。。。。。(3.29)