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