search

researchmm / Stark

[ICCV'21] Learning Spatio-Temporal Transformer for Visual Tracking
MIT License
649 stars 143 forks source link

How to get the score map? #51

Open hongsheng-Z opened 3 years ago

hongsheng-Z commented 3 years ago

Thanks for your great work, I want to know how to get the score map, the code only provides the calculation of conf_score。

MasterBin-IIAU commented 2 years ago

@hongsheng-Z Hi, STARK only predict a single score for indicating whether the current tracking state is reliable, rather than predicting a dense score map like previous siamese trackers. If you mean the attention score maps, you can refer to the visualization code provided by DETR (https://colab.research.google.com/github/facebookresearch/detr/blob/colab/notebooks/detr_attention.ipynb)