maite.protocols.object_detection.ObjectDetectionTarget#

class maite.protocols.object_detection.ObjectDetectionTarget(*args, **kwargs)[source]#

An object-detection target protocol.

This class is used to encode both predictions and ground-truth labels in the object detection problem.

Implementers must populate the following attributes:

Attributes:
boxesArrayLike

An array representing object detection boxes in a single image with x0, y0, x1, y1 format and shape (N_DETECTIONS, 4)

labelsArrayLike

An array representing the integer labels associated with each detection box of shape (N_DETECTIONS,)

scores: ArrayLike

An array representing the scores associated with each box (of shape (N_DETECTIONS,))