代码拉取完成,页面将自动刷新
摘自官网:https://docs.ultralytics.com/modes/train/#arguments
model=yolov8n.pt 例如:yolov8l.pt、yolov8n.yaml 等,用哪个模型,需要将对应的文件从model文件夹里移动到主目录
传入的model.yaml文件或者model.pt文件,用于构建网络和初始化,不同点在于只传入yaml文件的话参数会随机初始化
data=data.yaml
训练数据集的配置yaml文件,不要和模型的yaml弄混!
epochs=300
训练轮次数
patience=50
指定暂停训练观察轮数,默认50,如果50轮没有精度提升,模型会直接停止训练
batch=16
训练批次,默认16,如果内存或者显存不够,可适当调小
imgsz=640
训练图片大小,默认640
save=True
保存训练过程和训练权重,默认开启
save_period=-1
训练过程中每x个轮次保存一次训练模型,默认-1(不开启)
device=None
要运行的设备,即cuda device=0或Device=0,1,2,3或device=cpu
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。