from ultralytics import YOLO

# Load a model
model = YOLO('weights/yolov8n.pt')  # load an official model

model.train(data='datasets/safehat_2m/safehat.yaml', epochs=100)

model.val()