from ultralytics import YOLO
# Load a model
model = YOLO('yolov8n.pt') # load an official model
model = YOLO('C:/Users/Administrator/Desktop/ultralytics-main/runs/detect/train26/weights/best.pt') # load a custom model
# Predict with the model
results = model('C:/Users/Administrator/Desktop/ultralytics-main/datasets/mydata/images/0163.jpg') # predict on an image
# Process results list
for result in results:
boxes = result.boxes # Boxes object for bounding box outputs
masks = result.masks # Masks object for segmentation masks outputs
keypoints = result.keypoints # Keypoints object for pose outputs
probs = result.probs # Probs object for classification outputs
result.show() # display to screen
result.save(filename='result.jpg') # save to disk
model = YOLO('yolov8n.pt')
上述加载的是官方的模型,下述是我们自己训练好的模型文件(best.pt),写绝对路径(完整路径)。如下图
model = YOLO('C:/Users/Administrator/Desktop/ultralytics-main/runs/detect/train26/weights/best.pt') # load a custom model
接下来,就是我们需要预测的图像(0163.jpg),写绝对路径。如下图
results = model('C:/Users/Administrator/Desktop/ultralytics-main/datasets/mydata/images/0163.jpg') # predict on an image
后面部分便是候选框、掩码、关键点、物体类别的展示,通过show()函数来实现,save()函数保存该预测图像。