(video_rgb: np.ndarray, grid_points_px: np.ndarray)

(logits_or_scores: np.ndarray)

(frame_idx: int, points_world: np.ndarray)

(x_world: np.ndarray, t_cw: np.ndarray)

(x_world: np.ndarray, t_cw: np.ndarray)

(*args: Any, **kwargs: Any)

(self, base_seed: int)

(item: Any)

(self, query_tokens: torch.Tensor, memory_tokens: torch.Tensor)

(self, query_tokens: torch.Tensor, memory_tokens: torch.Tensor)

(self, batch: dict[str, Any])

(self, uv: torch.Tensor)

( self, video: torch.Tensor, u: torch.Tensor, v: torch.Tensor, t_src:

(self, decoded_queries: torch.Tensor)

(self, tokens: torch.Tensor)

(self, video_b_t_c_h_w: torch.Tensor, extra_tokens: torch.Tensor | None = None)

( self, outputs: dict[str, torch.Tensor], batch: dict[str, Any], )

()

(self)

()

(self, key: str)

(self, query_pool: dict[str, np.ndarray], rng: np.random.Generator)

(worker_id: int)

(self, epoch: int)

(self, worker_id: int)