Question about NMSFreeCoder

[george0407]

Hello,
I wonder why the code below use a global topk to select higher scores.
cls_scores = cls_scores.sigmoid() scores, indexs = cls_scores.view(-1).topk(max_num) labels = indexs % self.num_classes bbox_index = indexs // self.num_classes bbox_preds = bbox_preds[bbox_index]

What if there are 2 or more higher scores generated by one query?

[nacayu]

Hello, I have the same question as you, if you solved, please tell me. Thanks~

Hello, I wonder why the code below use a global topk to select higher scores. cls_scores = cls_scores.sigmoid() scores, indexs = cls_scores.view(-1).topk(max_num) labels = indexs % self.num_classes bbox_index = indexs // self.num_classes bbox_preds = bbox_preds[bbox_index]

What if there are 2 or more higher scores generated by one query?