Skip to content

amin-tohidi/custom_YOLOv8

 
 

Repository files navigation

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)

Use the model

model.train(data="coco128.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

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.3%
  • Other 0.7%