vis4d.op.detect.common

Common classes and functions for detection.

Classes

DetOut(boxes, scores, class_ids)

Output of the detection model.

class DetOut(boxes: list[Tensor], scores: list[Tensor], class_ids: list[Tensor])

Output of the detection model.

boxes (list[Tensor]): 2D bounding boxes of shape [N, 4] in xyxy format. scores (list[Tensor]): confidence scores of shape [N,]. class_ids (list[Tensor]): class ids of shape [N,].

Create new instance of DetOut(boxes, scores, class_ids)

boxes: list[Tensor]

Alias for field number 0

class_ids: list[Tensor]

Alias for field number 2

scores: list[Tensor]

Alias for field number 1