from ultralytics import YOLO


if __name__ == "__main__":
    model = YOLO('yolov12n.yaml')
    model.load("yolov12n.pt")

    # Train the model
    results = model.train(
    data='/Users/jsonxu/Desktop/yolov12/dataset/data.yaml',
    epochs=100, 
    batch=16, 
    imgsz=640,
    scale=0.5,  # S:0.9; M:0.9; L:0.9; X:0.9
    mosaic=1.0,
    mixup=0.0,  # S:0.05; M:0.15; L:0.15; X:0.2
    copy_paste=0.1,  # S:0.15; M:0.4; L:0.5; X:0.6
    )