【目标检测】|yolov5 yaml文件
depth_multiple: 0.33 # model depth multiple 控制模型的深度（BottleneckCSP个数）
width_multiple: 0.50 # layer channel multiple 控制Conv通道channel个数（卷积核数量）
# depth_multiple表示BottleneckCSP模块的缩放因子，将所有BottleneckCSP模块的Bottleneck乘上该参数得到最终个数。
# width_multiple表示卷积通道的缩放因子，就是将配置里面的backbone和head部分有关Conv通道的设置，全部乘以该系数。
# 通过这两个参数就可以实现不同复杂度的模型设计。
anchors:
- [10,13, 16,30, 33,23] # P3/8，检测小目标，10,13是一组尺寸，总共三组检测小目标,FPN接主干网络下采样8倍后的anchor大小
- [30,61, 62,45, 59,119] # P4/16，检测中目标，共三组,下采样4倍后的anchor大小
- [116,90, 156,198, 373,326] # P5/32，检测大目标，共三组,下采样2倍后的anchor大小
该anchor尺寸是为输入图像640×640分辨率预设的，实现了即可以在小特征图（feature map）上检测大目标，也可以在大特征图上检测小目标。三种尺寸的特征图，每个特征图上的格子有三个尺寸的anchor。
yolov5-5
backbone:
# [from, number, module, args]
[[-1, 1, Focus, [64, 3]], # 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, 9, 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, 1, SPP, [1024, [5, 9, 13]]],
[-1, 3, C3, [1024, False]], # 9
]
yolov5-6
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：输入来自那一层，-1代表上一层，1代表第1层，3代表第3层
number：与depth_multiple结合确定网络的深度，而且number数量要大于1
module：与width_multiple结合确定网络的宽度，主要是改变卷积核的数量
args：模块参数，channel，kernel_size，stride，padding，bias等
当第三个参数为 Focus 时，对特征图进行切片操作，[64,3]得到[3,32,3]，即输入channel=3（RGB），输出为64*0.5(width_multiple)=32，3为卷积核尺寸。 第四个参数中，第一个值为该模块中需要用到的通道数，第二个值为卷积核大小；
当第三个参数为 Conv 时，第四个参数中，第一个值为该模块中需要用到的通道数，第二个值为卷积核大小，第三个参数为步距大小，第四个是padding；nn.conv(kenel_size=1，stride=1，groups=1，bias=False) + Bn + Leaky_ReLu。
eg [-1, 1, Conv, [128, 3, 2]]：输入来自上一层，模块数量为1个，子模块为Conv，网络中最终有128*0.5(width_multiple)=32个卷积核，卷积核尺寸为3，stride=2,。
当第三个参数为 BottleneckCSP（C3） 时，第四个参数中，第一个值是该模块用到的通道数；如果存在第二个参数，第二个参数：是否启用 shortcut 连接. C3借鉴CSPNet网络结构，由3个卷积层和X个残差模块Concat组成，若有False，则没有残差模块，那么组成结构为nn.conv+Bn+Leaky_ReLu.
当第三个参数为 SPP时，第四个参数就是 SPP 中需要用到的卷积核大小。
当第三个参数为 nn.Upsample时，就是 torch 中实现的上采样函数。
当第三个参数为 Concat时，第四个参数就是 concat 中拼接的维度。
当第三个参数为 Detect时，第四个参数中，第一个值为类别个数，第二个值为超参数 anchors 的值。
YOLOV5-6 相比于YOLOV5-5更新了如下结构：
- 将Focus()模块，用一个Conv(k=6,s=2,p=2)代替
- 使用SPPF() 替换 SPP() 层以减少操作
- 减少 P3 主干层 C3() 从 9 到 6 重复以提高速度
- 重新排序将 SPPF() 放在主干的末尾
- 在最后一个 C3() 主干层中重新引入shortcut
- 更新了mixup和copy-paste 的超参
YOLOv5中的Head包括Neck和Detect_head两部分。Neck采用了PANet机构，Detect结构和YOLOv3中的Head一样。其中BottleNeckCSP带有False，说明没有使用残差结构，而是采用的backbone中的Conv。
head:
[[-1, 1, Conv, [512, 1, 1]],
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
[[-1, 6], 1, Concat, [1]], # cat backbone P4
[-1, 3, C3, [512, False]], # 13
[-1, 1, Conv, [256, 1, 1]],
[-1, 1, nn.Upsample, [None, 2, 'nearest']],
[[-1, 4], 1, Concat, [1]], # cat backbone P3
[-1, 3, C3, [256, False]], # 17 (P3/8-small)
[-1, 1, Conv, [256, 3, 2]],
[[-1, 14], 1, Concat, [1]], # cat head P4
[-1, 3, C3, [512, False]], # 20 (P4/16-medium)
[-1, 1, Conv, [512, 3, 2]],
[[-1, 10], 1, Concat, [1]], # cat head P5
[-1, 3, C3, [1024, False]], # 23 (P5/32-large)
[[17, 20, 23], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)
]
初始化超参
YOLOv5的超参文件见data/hyp.finetune.yaml（适用VOC数据集）或者hyo.scrach.yaml（适用COCO数据集）文件
lr0: 0.01 # 初始学习率 (SGD=1E-2, Adam=1E-3)
lrf: 0.2 # 循环学习率 (lr0 * lrf)
momentum: 0.937 # SGD momentum/Adam beta1 学习率动量
weight_decay: 0.0005 # 权重衰减系数
warmup_epochs: 3.0 # 预热学习 (fractions ok)
warmup_momentum: 0.8 # 预热学习动量
warmup_bias_lr: 0.1 # 预热初始学习率
box: 0.05 # iou损失系数
cls: 0.5 # cls损失系数
cls_pw: 1.0 # cls BCELoss正样本权重
obj: 1.0 # 有无物体系数(scale with pixels)
obj_pw: 1.0 # 有无物体BCELoss正样本权重
iou_t: 0.20 # IoU训练时的阈值
anchor_t: 4.0 # anchor的长宽比（长:宽 = 4:1）
# anchors: 3 # 每个输出层的anchors数量(0 to ignore)
#以下系数是数据增强系数，包括颜色空间和图片空间
fl_gamma: 0.0 # focal loss gamma (efficientDet default gamma=1.5)
hsv_h: 0.015 # 色调 (fraction)
hsv_s: 0.7 # 饱和度 (fraction)
hsv_v: 0.4 # 亮度 (fraction)
degrees: 0.0 # 旋转角度 (+/- deg)
translate: 0.1 # 平移(+/- fraction)
scale: 0.5 # 图像缩放 (+/- gain)
shear: 0.0 # 图像剪切 (+/- deg)
perspective: 0.0 # 透明度 (+/- fraction), range 0-0.001
flipud: 0.0 # 进行上下翻转概率 (probability)
fliplr: 0.5 # 进行左右翻转概率 (probability)
mosaic: 1.0 # 进行Mosaic概率 (probability)
mixup: 0.0 # 进行图像混叠概率（即，多张图像重叠在一起） (probability)
训练超参数包括：yaml文件的选择，和训练图片的大小，预训练，batch，epoch等。
可以直接在train.py的parser中修改，也可以在命令行执行时修改，如：$ python train.py --data coco.yaml --cfg yolov5s.yaml --weights ‘’ --batch-size 64
–data指定训练数据文件 --cfg设置网络结构的配置文件 –weihts加载预训练模型的路径
ref
https://zhuanlan.zhihu.com/p/172121380