1. 项目背景与核心价值
在目标检测模型的训练过程中,数据增强是提升模型泛化能力的关键手段。YOLOv5作为当前工业界最受欢迎的实时目标检测框架之一,其数据增强策略主要通过hyp.scratch.yaml配置文件进行集中管理。这个文件就像模型训练的"调味手册",决定了数据在输入网络前要经过哪些"烹饪处理"。
实际项目中常见这样的困境:明明使用了开源预训练权重,在自己的数据集上效果却不理想;尝试调整学习率等参数收效甚微,这时候就需要把注意力转向数据增强策略的调优。通过合理配置增强参数,我们可以在不增加标注成本的情况下,显著提升模型对小目标、遮挡、光照变化等复杂场景的适应能力。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 配置文件深度解析
2.1 文件结构与参数层级
hyp.scratch.yaml采用YAML格式组织,主要包含以下模块:
yaml复制# 超参数分组示意
lr0: 0.01 # 初始学习率
momentum: 0.937 # SGD动量
weight_decay: 0.0005 # 权重衰减
# 数据增强专用参数
hsv_h: 0.015 # 色调增强幅度
hsv_s: 0.7 # 饱和度增强幅度
hsv_v: 0.4 # 明度增强幅度
degrees: 10.0 # 旋转角度范围
translate: 0.1 # 平移比例
scale: 0.5 # 缩放比例
shear: 0.0 # 剪切变换幅度
perspective: 0.0 # 透视变换系数
flipud: 0.0 # 上下翻转概率
fliplr: 0.5 # 左右翻转概率
mosaic: 1.0 # mosaic增强概率
mixup: 0.0 # mixup增强概率
2.2 关键参数作用机理
HSV空间变换组:
- hsv_h:色调偏移范围(0-0.5),模拟不同光照色温
- hsv_s:饱和度缩放因子(0-1),控制颜色鲜艳程度
- hsv_v:明度缩放因子(0-1),模拟光照强度变化
几何变换组:
- degrees:随机旋转角度范围(建议0-45)
- translate
