Spaces:
Running
on
Zero
Running
on
Zero
File size: 571 Bytes
9b33fca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
"""Common classes and functions for tracking."""
from __future__ import annotations
from typing import NamedTuple
from torch import Tensor
class TrackOut(NamedTuple):
"""Output of track model.
Attributes:
boxes (list[Tensor]): List of bounding boxes (B, N, 4).
class_ids (list[Tensor]): List of class ids (B, N).
scores (list[Tensor]): List of scores (B, N).
track_ids (list[Tensor]): List of track ids (B, N).
"""
boxes: list[Tensor]
class_ids: list[Tensor]
scores: list[Tensor]
track_ids: list[Tensor]
|