yolov5超参数设置

### YOLOv5 超参数设置指南 在 YOLOv5 的训练过程中，超参数的合理配置对于模型性能至关重要。YOLOv5 提供了一个灵活的框架来调整各种超参数，这些参数通常存储在一个 YAML 文件中（例如 `hyp.scratch.yaml` 或自定义文件）。以下是关于如何进行超参数设置的具体说明： #### 1. 配置文件结构 超参数配置文件是一个 YAML 格式的文档，其中包含了诸如学习率、动量、权重衰减等关键参数。默认情况下，YOLOv5 使用的超参数文件位于项目的根目录下的 `data/hyps/` 文件夹内[^2]。 以下是一个典型的超参数配置文件示例： ```yaml lr0: 0.01 # 初始学习率 (SGD=1E-2, Adam=1E-3) lrf: 0.01 # 最终学习率因子 (lr0 * lrf) momentum: 0.937 # 动量优化器中的 Momentum 参数 weight_decay: 0.0005 # 权重衰减系数 (L2 正则化) warmup_epochs: 3.0 # 温暖启动周期数 warmup_momentum: 0.8 # 温暖启动期间使用的初始 Momentum 值 warmup_bias_lr: 0.1 # 温暖启动阶段偏置层的学习率 box: 0.05 # Box loss gain cls: 0.5 # Class loss gain obj: 1.0 # Objectness loss gain (scale with pixels) anchor_t: 4.0 # Anchor threshold value below which anchors are not used fl_gamma: 0.0 # Focal loss gamma hyperparameter (>=0 only) hsv_h: 0.015 # Image HSV-Hue augmentation (fraction) hsv_s: 0.7 # Image HSV-Saturation augmentation (fraction) hsv_v: 0.4 # Image HSV-Value augmentation (fraction) degrees: 0.0 # Image rotation (+/- deg) translate: 0.1 # Image translation (+/- fraction) scale: 0.5 # Image scale (+/- gain) shear: 0.0 # Image shear (+/- deg) perspective: 0.0 # Image perspective (+/- fraction), range 0-0.001 flipud: 0.0 # Image vertical flip probability fliplr: 0.5 # Image horizontal flip probability mosaic: 1.0 # Mosaic augmentation probability mixup: 0.0 # MixUp augmentation probability copy_paste: 0.0 # Copy-Paste augmentation probability ``` 上述配置文件涵盖了多个方面，包括但不限于优化器参数、损失函数增益以及数据增强选项。 #### 2. 自定义超参数 如果需要针对特定任务微调超参数，可以通过修改现有的 YAML 文件或将新的超参数文件传递给训练脚本实现。例如，在运行训练命令时可以指定不同的超参数文件路径： ```bash python train.py --img 640 --batch 16 --epochs 50 --data dataset.yaml --cfg yolov5s.yaml --weights '' --name custom_experiment --hyp hyp.custom.yaml ``` 在此命令中，通过 `--hyp` 参数指定了名为 `hyp.custom.yaml` 的自定义超参数文件。 #### 3. 数据增强与正则化 除了传统的优化器相关参数外，YOLOv5 还提供了丰富的数据增强功能以提高模型泛化能力。例如，HSV 变换、旋转、缩放和平移等操作都可以通过调整相应的超参数开启或关闭。 #### 4. 学习率调度策略 YOLOv5 默认采用余弦退火学习率调度器，并允许用户通过 `warmup_epochs` 和其他相关参数控制预热期的行为。这种设计有助于在网络初期更平稳地收敛。 ---