AssafSinger94 / dino-tracker

Official Pytorch Implementation for “DINO-Tracker: Taming DINO for Self-Supervised Point Tracking in a Single Video”
MIT License
361 stars 39 forks source link

How to achieve multi-target tracking with multiple targets #21

Closed Pdsn5 closed 1 month ago

Pdsn5 commented 1 month ago

Hello, does the algorithm support multi-target tracking? If so, how should it be implemented?

tnarek commented 1 month ago

hi @Pdsn5, please see inference_grid.py, where multiple query points are tracked in a single batch:

python ./inference_grid.py \
    --config ./config/train.yaml \
    --data-path <VIDEO_DIR_PATH> \
    --use-segm-mask # optional, used for sampling only foreground points