jozhang97 / DETA

Detection Transformers with Assignment
Apache License 2.0
243 stars 20 forks source link

first stage assignments thershold #25

Closed SangbumChoi closed 1 year ago

SangbumChoi commented 1 year ago

Hi, in your paper it says. We use a threshold 0.7 for the first stage.

but your code seems to use 0.9.

Also, is there any specific reason for keep top Q/L indices for L levels instead of ignoring the ratio? Thanks

jozhang97 commented 1 year ago

Thanks for pointing this out, we will update the paper.

Keeping the top Q/L makes sure each level has the same number of proposals per level. There is no specific reason besides it seems to work

SangbumChoi commented 1 year ago

Thanks for the reply!