search

naver / r2d2

Other
461 stars 86 forks source link

Visualize heatmap #19

Closed dzungdoan6 closed 3 years ago

dzungdoan6 commented 4 years ago

Hi, thank you for the significant work. I really enjoy your paper.

I would like to visualize the heatmap of repeatability and reliability (as Fig 6). Could you please guide me how to do it?

Thank you

jerome-revaud commented 4 years ago

Hi

Hmm good question. For the paper we wrote some code to generate figure 6, and I have to dig up a little to find it. [...] Ok, i found some old code which i cleaned a little bit. Here is how it works:

git pull
python viz_heatmaps.py --img imgs/peppers.png --checkpoint models/r2d2_WASF_N16.pt --gpu 0

You should see: viz