File size: 728 Bytes
13ed643 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from ultralytics import YOLO
# Load a model
# model = YOLO("yolov8n.yaml") # build a new model from scratch
# model = YOLO("yolov8n.pt") # load a pretrained model (recommended for training)
model = YOLO("yolov8m.yaml") # build a new model from scratch
model = YOLO("yolov8m.pt") # load a pretrained model (recommended for training)
# Use the model
# model.train(data="coco128.yaml", epochs=3) # train the model
model.train(data="./datasets/manga109/manga109.yaml", epochs=3) # train the model
metrics = model.val() # evaluate model performance on the validation set
# results = model("https://ultralytics.com/images/bus.jpg") # predict on an image
path = model.export(format="onnx") # export the model to ONNX format
|