search

liuan0803 / RADM

34 stars 0 forks source link

How to specify the number of generated bboxes #5

Closed naoto0804 closed 4 months ago

naoto0804 commented 8 months ago

Given n text features, RADM may generate a non-n number of elements for text. Is there any way to post-process or filter the outputs? I'm training on RADM on my dataset and facing the aforementioned issue.

liuan0803 commented 8 months ago

You can first use an NMS operation, then sort the results in descending order of confidence, and select the topN box according to your needs.