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()