Class Tracks

flowmap/tracking/track_predictor.py:14–20

@dataclass
class Tracks(Manipulable):
    xy: Float[Tensor, "batch frame point 2"]
    visibility: Bool[Tensor, "batch frame point"]

    # This is the first frame in the track sequence, not the query frame used to
    # generate the sequence, which is often different.
    start_frame: int


class TrackPredictor(nn.Module, ABC, Generic[T]):