深度学习中的目标检测与跟踪实战
1. YOLOv7 模型训练参数与监控
在使用 YOLOv7 模型进行训练时,有一系列重要的参数需要设置,这些参数会影响训练的过程和结果。以下是一些关键参数及其含义:
| 参数 | 含义 |
| ---- | ---- |
| --device 0,1,2,3 | 指定用于训练的 GPU,这里选择了 0、1、2、3 号 GPU。 |
| --sync-bn | 使用同步批量归一化,可提高分布式训练时的收敛性。 |
| --batch-size 16 | 每次迭代的批量大小为 16 张图像。 |
| --data /content/safety-Helmet-Reflective-Jacket/data.yaml | 指向包含数据集配置详情的 YAML 文件。 |
| --img 640 640 | 定义输入图像的大小为 640×640 像素。 |
| --cfg cfg/training/yolov7.yaml | 指定 YOLOv7 架构的模型配置文件(YAML)。 |
| --weights '' | 表示模型不使用预训练权重,从头开始训练。 |
| --name yolov7-ppe | 为训练会话提供一个名称,用于保存检查点和日志。 |
| --hyp data/hyp.scratch.p5.yaml |
超级会员免费看
订阅专栏 解锁全文
9855

被折叠的 条评论
为什么被折叠?



