# 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