RuntimeError: selected index k out of range

[SHUSHENGQIGUI]

when I input 7350X150 image for inference, or some images with high height-width ratio ， it will get this bug, the bug detial is in below:
File "/GroundingDINO/groundingdino/models/GroundingDINO/transformer.py", line 278, in forward
topk_proposals = torch.topk(topk_logits, topk, dim=1)[1] # bs, nq
RuntimeError: selected index k out of range

how to fix this bug ? please

[rentainhe]

After the data augmentation, the image size may be too small and there may not be enough feature for selection

[SHUSHENGQIGUI]

Thank you! do you know what is the minimal limit of the image size ratio?

[rentainhe]

Thank you! do you know what is the minimal limit of the image size ratio?

Sry we did not test some extreme situations before : (

I think Grounding-DINO works for most of the testing images

[artemisp]

Attempted a quick fix here: #346 (comment)
Feel free to try it out to help test it - or suggest a better solution. This is a really quick patch.