参数 | 说明 |
weights | 初始化的权重文件的路径地址 |
cfg | 模型配置文件的路径地址 |
data | 数据配置文件的路径地址 |
hyp | 超参数文件路径地址 |
epochs | 训练次数 |
batch-size | 批次大小 |
imgsz | 输入图片尺寸 |
rect | 是否采用矩形训练,默认False |
resume | 恢复训练 |
nosave | 不保存模型,只保存最后一轮,默认False |
noval | 不进行验证,只验证最后一轮,默认False |
noautoanchor | 不自动调整anchor,默认False |
evolve | 是否进行超参数进化,默认False |
bucket | 谷歌云盘bucket,一般不会用到 |
cache | 是否提前缓存图片到内存,以加快训练速度,默认False |
image-weights | 使用加权图像选择进行训练 |
device | 训练的设备,cpu;0(表示一个gpu设备cuda:0);0,1,2,3(多个gpu设备) |
multi-scale | 是否进行多尺度训练,默认False |
single-cls | 数据集是否只有一个类别,默认False |
optimizer | 优化器 |
sync-bn | 是否使用跨卡同步BN,在DDP模式使用 |
workers | 最大工作核心数 |
project | 训练模型的保存位置 |
name | 模型保存的目录名称 |
YOLO一些参数的意义
最新推荐文章于 2024-08-26 18:56:03 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)