Logo
Section 1 / 4

YOLO中的预训练模型与迁移学习

了解如何针对你的使用场景选择合适的训练方法。

导入

YOLO中的预训练模型与迁移学习

了解如何针对你的使用场景选择合适的训练方法。

from ultralytics import YOLO

从零开始构建一个新模型,使用预定义的yaml文件(如 yolov8n.yaml)

model = YOLO("yolov8n.yaml")

训练模型

model.train(data='coco8.yaml', epochs=10)

from ultralytics import YOLO

model = YOLO('yolov8n.pt') # 加载一个预训练的YOLOv8n检测模型 model.train(data='coco8.yaml', epochs=3) # 训练模型

提升模型性能:处理过拟合与欠拟合

微调与超参数简介