Yolo V8 CPU 训练 demo

#Yolo V8 CPU 训练 demo

from ultralytics import YOLO

def demo():
    model = YOLO('yolov8n.yaml')  # 不使用预训练权重训练
    # model = YOLO(r'yolov8p.yaml').load("yolov8n.pt")  # 使用预训练权重训练
    # Trainparameters ----------------------------------------------------------------------------------------------
    model.train(
        data='coco128.yaml',
        epochs=30,  # (int) number of epochs to train for
        patience=50,  # (int) epochs to wait for no observable improvement for early stopping of training
        batch=8,  # (int) number of images per batch (-1 for AutoBatch)
        imgsz=320,  # (int) size of input images as integer or w,h
        save=True,  # (bool) save train checkpoints and predict results
        save_period=-1,  # (int) Save checkpoint every x epochs (disabled if < 1)
        cache=False,  # (bool) True/ram, disk or False. Use cache for data loading
        # 如果是 GPU 则需修改
        device