minor bug in example code for background prediction

WalBouss / GEM

[CVPR24] Official Implementation of GEM (Grounding Everything Module)

MIT License

86 stars 4 forks source link

Open tian1327 opened 7 months ago

tian1327 commented 7 months ago

Thanks for the great work! There is a minor bug here: pred[probs < threshold] = 0 should be pred[max_prob < threshold] = 0.