前言

源码： . 链接: . 注释版全部项目文件已上传至GitHub: .

这个脚本存放的是yolov5的相关超参。

1、训练相关参数

# 1、训练相关参数
lr0: 0.01              # 初始学习率(SGD=1E-2, Adam=1E-3)
lrf: 0.2               # 最终学习率, 以one_cycle形式或者线性从lr0衰减至lr0 * lrf
momentum: 0.937        # SGD momentum/Adam beta1
weight_decay: 0.0005   # optimizer权重衰减系数 5e-4
warmup_epochs: 3.0     # 前3个epoch进行warmup
warmup_momentum: 0.8   # warmup初始化动量
warmup_bias_lr: 0.1    # warmup初始bias学习率

2、损失函数相关参数

# 2、损失函数相关参数
box: 0.05       # box iou损失系数
cls: 0.5        # cls分类损失系数
cls_pw: 1.0     # cls BCELoss正样本权重
obj: 1.0        # obj loss gain (scale with pixels)
obj_pw: 1.0     # obj BCELoss正样本权重
fl_gamma: 0.0   # focal loss gamma (efficientDet default gamma=1.5)

3、其他几个参数

# 3、其他几个参数
iou_t: 0.20     # 应该是很久以前使用的一个参数，因为那个时候筛选正负样本是靠iou来筛选的，这个iou_t应该就是筛选正负样本用的iou阈值，现在都用宽高比来筛选了，所以这个参数现在也没什么用了
anchor_t: 4.0   #  anchor的长宽比阈值（长:宽 = 4:1） 用于k-means中计算 bpr和aat
#anchors: 3     # 每个输出层的anchors数量 (0 to ignore)

4、数据增强相关参数

# 4、数据增强相关参数
hsv_h: 0.015      # hsv增强系数 色调
hsv_s: 0.7        # hsv增强系数 饱和度
hsv_v: 0.4        # hsv增强系数 亮度
degrees: 0.0      # random_perspective增强系数 旋转角度  (+/- deg)
translate: 0.1    # random_perspective增强系数 平移     (+/- fraction)
scale: 0.5        # random_perspective增强系数 图像缩放  (+/- gain)
shear: 0.0        # random_perspective增强系数 图像剪切  (+/- deg)
perspective: 0.0  # random_perspective增强系数 透明度    (+/- fraction), range 0-0.001
flipud: 0.0       # 上下翻转数据增强(probability)
fliplr: 0.5       # 左右翻转数据增强(probability)
mosaic: 1.0       # mosaic数据增强(probability)
mixup: 0.0        # mixup数据增强(probability)
cutout: 0.0       # cutout数据增强(probability)

Reference

王吉吉丫: .