Detections

inference_models.Detections dataclass

Attributes

bboxes_metadata class-attribute instance-attribute

bboxes_metadata = None

class_id instance-attribute

class_id

confidence instance-attribute

confidence

image_metadata class-attribute instance-attribute

image_metadata = None

xyxy instance-attribute

xyxy

Functions

__init__

__init__(xyxy, class_id, confidence, image_metadata=None, bboxes_metadata=None)

to_supervision

to_supervision()

Convert detections to Supervision Detections format.

Converts the PyTorch tensor-based detections to Supervision's NumPy-based format for visualization and analysis. This enables use of Supervision's rich ecosystem of annotators, trackers, and utilities.

Returns:

• Detections –

  sv.Detections: Supervision Detections object with:

  • xyxy: Bounding boxes as NumPy array (N, 4) in [x1, y1, x2, y2] format

  • class_id: Class IDs as NumPy array (N,)

  • confidence: Confidence scores as NumPy array (N,)

Examples:

Convert and visualize detections:

>>> import cv2
>>> import supervision as sv
>>> from inference_models import AutoModel
>>>
>>> model = AutoModel.from_pretrained("yolov8n-640")
>>> image = cv2.imread("image.jpg")
>>> predictions = model(image)
>>>
>>> # Convert to Supervision format
>>> detections = predictions[0].to_supervision()
>>>
>>> # Use Supervision annotators
>>> annotator = sv.BoxAnnotator()
>>> annotated = annotator.annotate(image.copy(), detections)

Filter by confidence:

>>> detections = predictions[0].to_supervision()
>>> high_conf = detections[detections.confidence > 0.7]

See Also

• Supervision documentation: https://supervision.roboflow.com