from ultralytics import YOLO

# Load an official or custom model
#model = YOLO('yolov8s-seg.pt')  # Load an official Detect model

model = YOLO('fivefingers-seg-100.pt')  # Load an official Detect model
results = model.train(data='data.yaml', epochs=100, imgsz=640, batch=8)