Method forward

flowmap/tracking/track_predictor.py:29–34

(
        self,
        videos: Float[Tensor, "batch frame 3 height width"],
        query_frame: int,
    )

    @abstractmethod
    def forward(
        self,
        videos: Float[Tensor, "batch frame 3 height width"],
        query_frame: int,
    ) -> Tracks:
        pass