from ultralytics import YOLO

# 加载预训练模型(可选:指定具体版本如yolov8n.pt)
model = YOLO("写模型文件")

# 启动训练(关键参数说明)
model.train(
    data="yolo-bvn.yaml",  # 使用绝对路径更安全,自己写的
    epochs=30,                # 增加epochs(小样本建议≥100)
    batch=10,                   # 根据CPU内存调整(若报错OOM则减小)
    workers=0,                 # 小样本无需多线程
    patience=20,               # 早停耐心值(防止过拟合)
    project="bvn_detection",   # 训练结果保存目录(自动生成)
    name="yolov8l01"                # 实验名称(可自定义)
)

可以帮跑训练,私聊联系

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐