From cbdb1d82ae747a6838480e32b070facea77ea923 Mon Sep 17 00:00:00 2001 From: zeyuzeng Date: Wed, 9 Feb 2022 18:28:54 +0800 Subject: [PATCH 1/2] [Fix] overflow bug --- mmdet3d/core/post_processing/box3d_nms.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mmdet3d/core/post_processing/box3d_nms.py b/mmdet3d/core/post_processing/box3d_nms.py index 638c9aaae0..314a138d0b 100644 --- a/mmdet3d/core/post_processing/box3d_nms.py +++ b/mmdet3d/core/post_processing/box3d_nms.py @@ -217,4 +217,7 @@ def circle_nms(dets, thresh, post_max_size=83): # ovr = inter / areas[j] if dist <= thresh: suppressed[j] = 1 - return keep[:post_max_size] + + if post_max_size < len(keep): + return keep[:post_max_size] + return keep From a76344dd9d72873453fb50eedeace67551ee44fd Mon Sep 17 00:00:00 2001 From: zeyu-hello Date: Fri, 11 Feb 2022 10:35:41 +0800 Subject: [PATCH 2/2] [Fix] typo --- mmdet3d/core/post_processing/box3d_nms.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mmdet3d/core/post_processing/box3d_nms.py b/mmdet3d/core/post_processing/box3d_nms.py index 314a138d0b..2a6459ce78 100644 --- a/mmdet3d/core/post_processing/box3d_nms.py +++ b/mmdet3d/core/post_processing/box3d_nms.py @@ -220,4 +220,5 @@ def circle_nms(dets, thresh, post_max_size=83): if post_max_size < len(keep): return keep[:post_max_size] + return keep