from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n-seg.pt')  # build a new model from YAML


# Train the model
results = model.train(data='seg_magnet.yaml', epochs=150, imgsz=640)