change nms thresholds on cpu only code

mlcommons · Jan 21, 2025 · dbf6eb4 · dbf6eb4
1 parent b63ca20
commit dbf6eb4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/automotive/3d-object-detection/model/pointpillars_core.py b/automotive/3d-object-detection/model/pointpillars_core.py
@@ -470,9 +470,9 @@ class PointPillarsPos(nn.Module):
     def __init__(self, nclasses=3):
         super().__init__()
         self.nclasses = nclasses
-        self.nms_thr = 0.01
+        self.nms_thr = 0.25
         self.score_thr = 0.1
-        self.max_num = 50
+        self.max_num = 500
 
     def nms_filter(self, bbox_pred, bbox_cls_pred, bbox_dir_cls_pred):
         # 3. nms