xingyizhou / GTR

Global Tracking Transformers, CVPR 2022

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Question: What is `data` in gtr/tracking/trackeval/metrics `eval_sequence` methods?

alckasoc opened this issue · comments

I don't have an issue. I'm just curious what the data variable has in https://github.com/xingyizhou/GTR/tree/master/gtr/tracking/trackeval/metrics eval_sequence methods.

From looking at clear.py, I'm guessing:

data = {
    "num_gt_ids": Total number of unique gt ids in the entire sequence,
    "num_tracker_dets": Total number of tracker detections in the entire sequence,
    "num_gt_dets": Total number of gt detections in the entire sequence,
    "gt_ids": [
        [0, 1],  # Zeroth frame.
        [0, 2, 3],
        ...
    ],
     "tracker_ids": [
        [0, 1, 2],  # Zeroth frame.
        [0, 2],
        ... 
    ],
    "similarity_scores": [  # Length is len(video) - 1
         # What is this?
    ]
}

Is similarity scores just a list of these: traj_score = torch.mm(asso_nonk, id_inds) # n_k x M?

It looks like the similarity score matrix is of shape: (number of ground truth instances at frame t, number of tracked instances at frame t). I'm still confused about what this exactly is.

Is there code provided for getting this data format (i.e. from instances returned from the sliding_inference method here)?

Thank you!