导入
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) # 训练模型