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