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

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐