YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
from ultralytics import YOLO
Load a model
model = YOLO("yolov8n.pt")
Train the model
train_results = model.train( data="coco8.yaml", # path to dataset YAML epochs=100, # number of training epochs imgsz=640, # training image size device="cpu", # device to run on, i.e. device=0 or device=0,1,2,3 or device=cpu )
Evaluate model performance on the validation set
metrics = model.val()
Perform object detection on an image
results = model("path/to/image.jpg") results[0].show()
Export the model to ONNX format
path = model.export(format="onnx") # return path to exported model