yolov8_object_detection_with_supervision_bounding_box_annotation.py

1import cv2
2import supervision as sv
3from ultralytics import YOLO
4
5# load the model
6model = YOLO('yolov8n.pt')
7
8# load the image
9image = cv2.imread("image.jpg")
10
11# perform inference
12results = model(image)[0]
13
14# convert results to supervision Detections
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)