Source code for vis4d.op.typing

"""Vis4D op typing."""

from __future__ import annotations

from typing import NamedTuple

from torch import Tensor



[docs]
class Proposals(NamedTuple):
    """Output structure for 2D bounding box proposals."""

    boxes: list[Tensor]
    scores: list[Tensor]




[docs]
class Targets(NamedTuple):
    """Output structure for targets."""

    boxes: list[Tensor]
    classes: list[Tensor]
    labels: list[Tensor]