megvii-research / MOTRv2

[CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors
Other
349 stars 44 forks source link

score threshold and update_score_threshold #15

Closed hahapt closed 1 year ago

hahapt commented 1 year ago

I found the score threshold and update_score_threshold is set to 0.5 in submit_dance.py, then the detections under 0.5 confidence will directly be removed.
So why it set to be 0.5? Is there any hyper-parameter during training influence them?

zyayoung commented 1 year ago

We did a grid search on the score threshold and update_score_threshold during inference.

A corresponding threshold in training is also set to 0.5. https://github.com/megvii-research/MOTRv2/blob/be49b7336218e470c9ebcd34be54fe7eec702675/models/qim.py#L70