yolov5hyp.scrath.low
时间: 2025-02-01 14:10:51 浏览: 41
### YOLOv5低参数配置
对于YOLOv5从零开始配置低参数(`hyp.scratch.low.yaml`),主要目的是为了减少模型训练所需的资源,提高训练效率并适应小型数据集或计算能力有限的情况。以下是具体的配置方法:
#### 配置文件结构
在YOLOv5项目中,默认提供了多种超参配置文件位于 `data/hyps/` 文件夹下。创建一个新的配置文件命名为 `hyp.scratch.low.yaml` 并调整如下关键参数[^1]:
```yaml
# YOLOv5 Scratch Hypers (low parameter setting)
lr0: 0.01 # 初始学习率 (SGD=1E-2, Adam=1E-3)
lrf: 0.1 # 最终学习率 = lr0 * lrf
momentum: 0.937 # SGD 动量/Adam beta1
weight_decay: 0.0005 # 权重衰减 (L2正则化系数)
warmup_epochs: 3.0 # 温和启动周期数
warmup_momentum: 0.8 # 温和启动动量
warmup_bias_lr: 0.1 # 温和启动偏置层初始学习率
box: 0.05 # box loss gain
cls: 0.5 # cls loss gain
cls_pw: 1.0 # cls BCELoss positive_weight
obj: 1.0 # obj loss gain (scale with pixels)
obj_pw: 1.0 # obj BCELoss positive_weight
iou_t: 0.20 # IoU training threshold
anchor_t: 4.0 # anchor-multiple threshold
fl_gamma: 0.0 # focal loss gamma (efficientDet default gamma=1.5)
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 # 图像旋转角度范围 (-rotate, +rotate)
translate: 0.1 # 图像平移比例 (+/- fraction)
scale: 0.5 # 图像缩放比例 (+/- gain)
shear: 0.0 # 剪切变换程度
perspective: 0.0 # 透视变换程度
flipud: 0.0 # 上下翻转概率
fliplr: 0.5 # 左右翻转概率
mosaic: 1.0 # 使用马赛克增强的概率
mixup: 0.0 # 使用混合样本增强的概率
copy_paste: 0.0 # 使用复制粘贴增强的概率
```
此配置降低了多项增广操作的程度,并减少了损失函数中的权重因子,使得整个网络更容易收敛,在较小的数据集上表现更好。
#### 训练命令示例
使用上述自定义的超参数配置文件进行训练时,可以通过指定路径来加载该文件:
```bash
python train.py --img 640 --batch 16 --epochs 100 --data custom_data.yaml --weights yolov5s.pt --cfg models/yolov5s.yaml --hyp data/hyps/hyp.scratch.low.yaml
```
阅读全文
相关推荐


















