yolov5 train.py 参数解析

1,013 阅读2分钟

opt参数解析

参数help帮助
--weightinitial weights path加载的权重文件路径
--cfgmodel.yaml path模型配置文件,网络结构 路径
--datadataset.yaml path数据集配置文件,数据集路径
--hyphyperparameters path超参数文件 路径
--epochsTotal training rounds训练总轮次
--batch-sizetotal batch size for all GPUs, -1 for autobatch批次大小
--imgsztrain, val image size (pixels)输入图片分辨率大小
--rectrectangular training是否采用矩形训练,默认False
--resumeresume most recent training接着打断训练上次的结果接着训练
--nosaveonly save final checkpoint只保存最终的模型,默认False
--noautoanchordisable AutoAnchor不自动调整anchor,默认False
--noplotssave no plot files不保存打印文件,默认False
--evolveevolve hyperparameters for x generations是否进行超参数进化,默认False
--bucketgsutil bucket谷歌云盘bucket,一般不会用到
--cache--cache images in "ram" (default) or "disk"是否提前缓存图片到内存,以加快训练速度,默认False
--devicecuda device, i.e. 0 or 0,1,2,3 or cpu训练的设备,cpu;0(表示一个gpu设备cuda:0);0,1,2,3(多个gpu设备)
--multi-scalevary img-size +/- 50%%是否进行多尺度训练,默认False
--single-clstrain multi-class data as single-class数据集是否只有一个类别,默认False
--optimizeroptimizer优化器
--sync-bnuse SyncBatchNorm, only available in DDP mode是否使用跨卡同步BN,在DDP模式使用
--workersmax dataloader workers (per RANK in DDP mode)dataloader的最大worker数量
--projectsave to project path保存到项目结果地址
--namesave to project/name/保存到项目结果/名称
--exist-okexisting project/name ok, do not increment现有项目/名称确定,不递增,默认False
--quadquad dataloader四元数据加载器 开启之后在尺寸大于640的图像上识别效果更好,但是有可能会使在640尺寸的图片上效果更差
--cos-lrcosine LR scheduler是否采用退火余弦学习率,默认False
--label-smoothingLabel smoothing epsilon标签平滑
--patienceEarlyStopping patience (epochs without improvement)早停机制,默认False
--freezFreeze layers: backbone=10, first3=0 1 2冻结层数,默认不冻结
--save-periodSave checkpoint every x epochs (disabled if < 1)用于记录训练日志信息,int 型,默认 -1
--seedGlobal training seed随机数设置
--local_rankAutomatic DDP Multi-GPU argument, do not modify自动单机多卡训练 一般不改动

看到这,,,好人顺便帮忙点赞关注哈。