Source code for vision3d.metrics._types

"""Shared types for :mod:`vision3d.metrics`."""

from typing import TypedDict

from torch import Tensor

from vision3d.tensors import BoundingBoxes3D




class Prediction3D(TypedDict):
    """Per-frame detection output.

    Attributes:
        boxes: ``[N, K]`` predicted 3D bounding boxes; ``K`` depends on
            the box format.
        scores: ``[N]`` confidence scores.
        labels: ``[N]`` integer class labels.
    """

    boxes: BoundingBoxes3D
    scores: Tensor
    labels: Tensor





class Target3D(TypedDict):
    """Per-frame ground-truth annotations.

    Attributes:
        boxes: ``[M, K]`` ground-truth 3D bounding boxes.
        labels: ``[M]`` integer class labels.
    """

    boxes: BoundingBoxes3D
    labels: Tensor