11 lines
248 B
Python
11 lines
248 B
Python
from ultralytics import YOLO
|
|
|
|
# Load a pre-trained YOLOv10n model
|
|
model = YOLO("./yolov10x.pt")
|
|
model.train(data="./dataset.yaml", epochs=3)
|
|
|
|
# Perform object detection on an image
|
|
results = model("./1.jpg")
|
|
|
|
# Display the results
|
|
results[0].show() |