From 28f4ae30ce1f1913f976296e2129b7e5e5d1970a Mon Sep 17 00:00:00 2001
From: jiangwei <1131066623@qq.com>
Date: Sat, 4 Jan 2025 09:35:14 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dfilter=5Fscores=5Fand=5Ftopk?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 mmdet/models/utils/misc.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmdet/models/utils/misc.py b/mmdet/models/utils/misc.py
index 2cf429153ba..743ad013f08 100644
--- a/mmdet/models/utils/misc.py
+++ b/mmdet/models/utils/misc.py
@@ -333,7 +333,7 @@ def filter_scores_and_topk(scores, score_thr, topk, results=None):
     scores = scores[valid_mask]
     valid_idxs = torch.nonzero(valid_mask)
 
-    num_topk = min(topk, valid_idxs.size(0))
+    num_topk = min(topk if topk != -1 else valid_idxs.size(0), valid_idxs.size(0))
     # torch.sort is actually faster than .topk (at least on GPUs)
     scores, idxs = scores.sort(descending=True)
     scores = scores[:num_topk]