yolov8_inference_with_supervision_box_label_annotation.py

1import cv2
2import supervision as sv
3from ultralytics import YOLO
4
5# Load a pre-trained model
6model = YOLO('yolov8n.pt')
7
8# Load an image
9image = cv2.imread("image.jpg")
10
11# Run inference
12results = model(image)[0]
13
14# Convert results to supervision Detections format
15detections = sv.Detections.from_ultralytics(results)
16
17# Create annotators
18box_annotator = sv.BoxAnnotator()
19label_annotator = sv.LabelAnnotator()
20
21# Annotate the image
22annotated_image = box_annotator.annotate(
23    scene=image.copy(),
24    detections=detections
25)
26annotated_image = label_annotator.annotate(
27    scene=annotated_image,
28    detections=detections
29)
30
31# Display the image
32sv.plot_image(annotated_image)