Views
No views yet
Predictor class from the official repository. Ensure you have the repository cloned and dependencies installed.1import torch
2from model.trackon_predictor import Predictor
3
4device = "cuda" if torch.cuda.is_available() else "cpu"
5
6# Initialize
7model = Predictor(checkpoint_path="path/to/checkpoint.pth").to(device).eval()
8
9# Inputs
10# video: (1, T, 3, H, W) in range 0-255
11# queries: (1, N, 3) with rows = (t, x, y) in pixel coordinates
12# or use None to enable the model's uniform grid querying
13video = ... # e.g., torchvision.io.read_video -> (T, H, W, 3) -> (T, 3, H, W) -> add batch dim
14queries = ... # e.g., torch.tensor([[0, 190, 190], [0, 200, 190], ...]).unsqueeze(0).to(device)
15
16# Inference
17traj, vis = model(video, queries)
18
19# Outputs
20# traj: (1, T, N, 2) -> per-point (x, y) in pixels
21# vis: (1, T, N) -> per-point visibility in {0, 1}huggingface-cli login), the weights will be automatically downloaded and cached locally on the first run.1@inproceedings{aydemir2026trackonr,
2 title = {Real-World Point Tracking with Verifier-Guided Pseudo-Labeling},
3 author = {Aydemir, G\"orkay and G\"uney, Fatma and Xie, Weidi},
4 booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
5 year = {2026}
6}