first
This commit is contained in:
+50
@@ -0,0 +1,50 @@
|
||||
# YOLOv8 训练配置模板
|
||||
# 改完后在项目根目录执行:
|
||||
# yolo train cfg=train.yaml
|
||||
|
||||
task: detect
|
||||
mode: train
|
||||
model: yolov8n.pt # n/s/m/l/x,或本地 best.pt
|
||||
data: 01data.yaml # 指向本项目数据集配置
|
||||
|
||||
epochs: 100
|
||||
batch: 8 # 显存不够改 4 或 2,-1 为自动
|
||||
imgsz: 640
|
||||
device: 0 # 0=GPU,cpu=CPU
|
||||
workers: 4 # Windows 报错可改 0
|
||||
amp: true
|
||||
|
||||
project: runs
|
||||
name: yolov8_gongjian
|
||||
exist_ok: true
|
||||
save: true
|
||||
save_period: 10
|
||||
plots: true
|
||||
val: true
|
||||
patience: 50
|
||||
|
||||
optimizer: auto # auto / SGD / Adam / AdamW
|
||||
lr0: 0.01
|
||||
lrf: 0.01
|
||||
momentum: 0.937
|
||||
weight_decay: 0.0005
|
||||
warmup_epochs: 3.0
|
||||
cos_lr: false
|
||||
close_mosaic: 10
|
||||
pretrained: true
|
||||
resume: false
|
||||
seed: 0
|
||||
|
||||
hsv_h: 0.015
|
||||
hsv_s: 0.7
|
||||
hsv_v: 0.4
|
||||
degrees: 0.0
|
||||
translate: 0.1
|
||||
scale: 0.5
|
||||
fliplr: 0.5
|
||||
mosaic: 1.0
|
||||
mixup: 0.0
|
||||
|
||||
box: 7.5
|
||||
cls: 0.5
|
||||
dfl: 1.5
|
||||
Reference in New Issue
Block a user