About top-k filtering - Githubissues

hkchengrex / Mask-Propagation

[CVPR 2021] MiVOS - Mask Propagation module. Reproduced STM (and better) with training code :star2:. Semi-supervised video object segmentation evaluation.

https://hkchengrex.github.io/MiVOS/

MIT License

128 stars 22 forks source link

About top-k filtering #46

Closed csustYyh closed 1 year ago

csustYyh commented 1 year ago

I am sorry to bother you. When I used top-k, I found a problem that my input values might be large sometimes. After 'values.exp_()', there will be 'INF'. To avoid overflow, I change the computation as: Is it reasonable to make this change?