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